|
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...
|
|