Home
last modified time | relevance | path

Searched defs:tensor_info (Results 1 – 25 of 37) sorted by relevance

12

/external/ComputeLibrary/tests/validation/UNIT/
DSubTensorInfo.cpp77 TensorInfo tensor_info(TensorShape(23U, 17U, 3U), 1, DataType::F32); in TEST_CASE() local
84 TensorInfo tensor_info(TensorShape(23U, 17U, 3U), 1, DataType::F32); in TEST_CASE() local
92 TensorInfo tensor_info(TensorShape(23U, 17U, 3U), 1, DataType::F32); in TEST_CASE() local
/external/tensorflow/tensorflow/lite/python/
Dconvert_saved_model.py33 def _log_tensor_details(tensor_info): argument
107 def gather_names(tensor_info): argument
/external/ComputeLibrary/src/core/CPP/
DValidate.h41 const ITensorInfo *tensor_info) in error_on_unsupported_cpu_fp16()
61 const ITensorInfo *tensor_info) in error_on_unsupported_cpu_bf16()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/
Dgreedy_by_breadth_assignment.cc59 for (const auto& tensor_info : task_profiles[task_id]) { in GreedyByBreadthAssignment() local
74 for (const auto& tensor_info : task_profiles[task.task_id]) { in GreedyByBreadthAssignment() local
/external/tensorflow/tensorflow/python/saved_model/
Dutils_impl.py156 def get_tensor_from_tensor_info(tensor_info, graph=None, import_scope=None): argument
199 def get_element_from_tensor_info(tensor_info, graph=None, import_scope=None): argument
Dsaved_model_test.py127 def _validate_inputs_tensor_info_fail(self, builder, tensor_info): argument
140 def _validate_inputs_tensor_info_accept(self, builder, tensor_info): argument
149 def _validate_outputs_tensor_info_fail(self, builder, tensor_info): argument
161 def _validate_outputs_tensor_info_accept(self, builder, tensor_info): argument
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEGaussian5x5.cpp47 TensorInfo tensor_info(input->info()->tensor_shape(), 1, DataType::S16); in configure() local
DNEFastCorners.cpp67 TensorInfo tensor_info(input->info()->tensor_shape(), Format::U8); in configure() local
DNESobel7x7.cpp52 TensorInfo tensor_info(input->info()->tensor_shape(), Format::S32); in configure() local
DNENormalizationLayer.cpp48 TensorInfo tensor_info(input->info()->tensor_shape(), 1, input->info()->data_type()); in configure() local
DNESobel5x5.cpp52 TensorInfo tensor_info(input->info()->tensor_shape(), Format::S16); in configure() local
DNEOpticalFlow.cpp104 TensorInfo tensor_info(TensorShape(width_ith, height_ith), Format::S16); in configure() local
DNEHOGMultiDetection.cpp200 TensorInfo tensor_info(*(multi_hog->model(idx_multi_hog)->info()), width, height); in configure() local
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Dembedder.cc167 TensorInfo tensor_info; in MakeModel() local
212 bool intermediate_only, TensorInfo* tensor_info) { in Tensors()
437 bool intermediate_only, const TensorInfo& tensor_info) { in Operators()
503 bool intermediate_only, TensorInfo* tensor_info) { in SubGraphs()
526 bool intermediate_only, const TensorInfo& tensor_info, in Buffers()
/external/ComputeLibrary/utils/
DImageLoader.h355 virtual void validate_info(const ITensorInfo *tensor_info) in validate_info()
414 void validate_info(const ITensorInfo *tensor_info) override in validate_info()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLSobel7x7.cpp62 TensorInfo tensor_info(input->info()->tensor_shape(), 1, DataType::S32); in configure() local
DCLSobel5x5.cpp62 TensorInfo tensor_info(input->info()->tensor_shape(), 1, DataType::S16); in configure() local
DCLFastCorners.cpp73 TensorInfo tensor_info(input->info()->tensor_shape(), 1, DataType::U8); in configure() local
DCLHarrisCorners.cpp89 TensorInfo tensor_info(shape, 1, dt); in configure() local
DCLOpticalFlow.cpp128 TensorInfo tensor_info(TensorShape(width_ith, height_ith), 1, DataType::S16); in configure() local
/external/ComputeLibrary/src/runtime/
DPyramid.cpp66 TensorInfo tensor_info(tensor_shape, _info.format()); in internal_init() local
/external/ComputeLibrary/src/runtime/CL/
DCLPyramid.cpp76 TensorInfo tensor_info(tensor_shape, _info.format()); in internal_init() local
/external/ComputeLibrary/src/core/CL/kernels/
DCLTransposeKernel.cpp66 …const TensorInfo tensor_info = input->clone()->set_tensor_shape(transposed_tensor_shape(input->ten… in validate_arguments() local
/external/tensorflow/tensorflow/core/tfrt/saved_model/
Dsaved_model.cc511 const tensorflow::TensorInfo& tensor_info = p.second; in GetSignaturesFromSignatureDef() local
525 const tensorflow::TensorInfo& tensor_info = p.second; in GetSignaturesFromSignatureDef() local
842 const auto& tensor_info = signature_def.inputs().at(input_names[j]); in RunMultipleSignatures() local
970 const auto& tensor_info = iter.second; in JoinSignatures() local
1009 const auto& tensor_info = signature_def.outputs().at(output_key); in JoinSignatures() local
/external/tensorflow/tensorflow/core/protobuf/
Dmeta_graph.proto337 TensorInfo tensor_info = 1; field

12