Compute Library
 21.08
GraphUtils.h File Reference
#include "arm_compute/core/PixelValue.h"
#include "arm_compute/core/Utils.h"
#include "arm_compute/core/utils/misc/Utility.h"
#include "arm_compute/graph/Graph.h"
#include "arm_compute/graph/ITensorAccessor.h"
#include "arm_compute/graph/Types.h"
#include "arm_compute/runtime/Tensor.h"
#include "utils/CommonGraphOptions.h"
#include <array>
#include <random>
#include <string>
#include <vector>

Go to the source code of this file.

Data Structures

class  IPreprocessor
 Preprocessor interface. More...
 
class  CaffePreproccessor
 Caffe preproccessor. More...
 
class  TFPreproccessor
 TF preproccessor. More...
 
class  PPMWriter
 PPM writer class. More...
 
class  DummyAccessor
 Dummy accessor class. More...
 
class  NumPyAccessor
 NumPy accessor class. More...
 
class  SaveNumPyAccessor
 SaveNumPy accessor class. More...
 
class  PrintAccessor
 Print accessor class. More...
 
class  ImageAccessor
 Image accessor class. More...
 
class  ValidationInputAccessor
 Input Accessor used for network validation. More...
 
class  ValidationOutputAccessor
 Output Accessor used for network validation. More...
 
class  DetectionOutputAccessor
 Detection output accessor class. More...
 
class  TopNPredictionsAccessor
 Result accessor class. More...
 
class  RandomAccessor
 Random accessor class. More...
 
class  NumPyBinLoader
 Numpy Binary loader class. More...
 

Namespaces

 arm_compute
 Copyright (c) 2017-2021 Arm Limited.
 
 arm_compute::graph_utils
 

Functions

std::unique_ptr< graph::ITensorAccessor > get_random_accessor (PixelValue lower, PixelValue upper, const std::random_device::result_type seed=0)
 Generates appropriate random accessor. More...
 
std::unique_ptr< graph::ITensorAccessor > get_weights_accessor (const std::string &path, const std::string &data_file, DataLayout file_layout=DataLayout::NCHW)
 Generates appropriate weights accessor according to the specified path. More...
 
std::unique_ptr< graph::ITensorAccessor > get_input_accessor (const arm_compute::utils::CommonGraphParams &graph_parameters, std::unique_ptr< IPreprocessor > preprocessor=nullptr, bool bgr=true)
 Generates appropriate input accessor according to the specified graph parameters. More...
 
std::unique_ptr< graph::ITensorAccessor > get_output_accessor (const arm_compute::utils::CommonGraphParams &graph_parameters, size_t top_n=5, bool is_validation=false, std::ostream &output_stream=std::cout)
 Generates appropriate output accessor according to the specified graph parameters. More...
 
std::unique_ptr< graph::ITensorAccessor > get_detection_output_accessor (const arm_compute::utils::CommonGraphParams &graph_parameters, std::vector< TensorShape > tensor_shapes, bool is_validation=false, std::ostream &output_stream=std::cout)
 Generates appropriate output accessor according to the specified graph parameters. More...
 
std::unique_ptr< graph::ITensorAccessor > get_npy_output_accessor (const std::string &npy_path, TensorShape shape, DataType data_type, DataLayout data_layout=DataLayout::NCHW, std::ostream &output_stream=std::cout)
 Generates appropriate npy output accessor according to the specified npy_path. More...
 
std::unique_ptr< graph::ITensorAccessor > get_save_npy_output_accessor (const std::string &npy_name, const bool is_fortran=false)
 Generates appropriate npy output accessor according to the specified npy_path. More...
 
std::unique_ptr< graph::ITensorAccessor > get_print_output_accessor (std::ostream &output_stream=std::cout)
 Generates print tensor accessor. More...
 
TensorShape permute_shape (TensorShape tensor_shape, DataLayout in_data_layout, DataLayout out_data_layout)
 Permutes a given tensor shape given the input and output data layout. More...
 
graph::Target set_target_hint (int target)
 Utility function to return the TargetHint. More...