/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/utils/ |
H A D | Utils.h | 38 inline bool is_alloc_tensor(const ITensorInfo *tensor_info) in is_alloc_tensor() 45 inline bool is_noalloc_tensor(const ITensorInfo *tensor_info) in is_noalloc_tensor() 52 inline bool is_valid_tensor(const ITensorInfo *tensor_info) in is_valid_tensor() 59 inline bool is_invalid_tensor(const ITensorInfo *tensor_info) in is_invalid_tensor()
|
/aosp_15_r20/external/ComputeLibrary/tests/validation/UNIT/ |
H A D | SubTensorInfo.cpp | 78 TensorInfo tensor_info(TensorShape(23U, 17U, 3U), 1, DataType::F32); in TEST_CASE() local 85 TensorInfo tensor_info(TensorShape(23U, 17U, 3U), 1, DataType::F32); in TEST_CASE() local 93 TensorInfo tensor_info(TensorShape(23U, 17U, 3U), 1, DataType::F32); in TEST_CASE() local 102 TensorInfo tensor_info(TensorShape(23U, 17U, 3U), 1, DataType::F32); in TEST_CASE() local
|
H A D | TensorInfo.cpp | 190 TensorInfo tensor_info(TensorShape(23U, 17U, 3U), 1, DataType::F32); in TEST_CASE() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/model_modifier/ |
H A D | validation_graph_builder.cc | 76 TensorInfo tensor_info; in MakeModel() local 124 TensorInfo* tensor_info) { in Tensors() 348 const TensorInfo& tensor_info) { in Operators() 412 TensorInfo* tensor_info) { in SubGraphs() 435 const TensorInfo& tensor_info, in Buffers()
|
/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/ |
H A D | PReluLayer.cpp | 98 const auto tensor_info = TensorInfo(TensorShape(32U, 13U, 2U), 1, DataType::F32); in TEST_CASE() local 103 SimpleTensor<float> compute_float_reference(const TensorInfo &tensor_info) in compute_float_reference() 131 const auto tensor_info = TensorInfo(TensorShape(33U, 13U, 2U), 1, DataType::F32); in TEST_CASE() local 141 const auto tensor_info = TensorInfo(TensorShape(33U, 13U, 2U), 1, DataType::F32); in TEST_CASE() local
|
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/ |
H A D | GpuKernelArgument.h | 80 GpuKernelArgument(const ITensorInfo &tensor_info, in GpuKernelArgument() 92 ITensorInfo *tensor_info() in tensor_info() function 97 const ITensorInfo *tensor_info() const in tensor_info() function
|
H A D | GpuWorkloadSourceCode.h | 55 GpuWorkloadArgument(const ITensorInfo &tensor_info, in GpuWorkloadArgument() 69 ITensorInfo *tensor_info() in tensor_info() function 74 const ITensorInfo *tensor_info() const in tensor_info() function
|
H A D | GpuWorkloadSketch.cpp | 46 void GpuWorkloadSketch::register_new_tensor(ITensorInfo &tensor_info) in register_new_tensor() 55 TensorInfo tensor_info{}; in create_tensor_info() local
|
H A D | GpuWorkloadSketchImpl.h | 129 ITensorInfo *create_auxiliary_tensor(const ITensorInfo &tensor_info) in create_auxiliary_tensor()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/ |
H A D | saved_model.cc | 402 const tensorflow::TensorInfo& tensor_info = p.second; in GetSignaturesFromSignatureDef() local 416 const tensorflow::TensorInfo& tensor_info = p.second; in GetSignaturesFromSignatureDef() local 794 const auto& tensor_info = signature_def.inputs().at(input_names[j]); in RunMultipleSignatures() local 815 const auto& tensor_info = signature_def.outputs().at(output_key); in RunMultipleSignatures() local 912 const auto& tensor_info = iter.second; in JoinSignatures() local 951 const auto& tensor_info = signature_def.outputs().at(output_key); in JoinSignatures() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
H A D | embedder.cc | 167 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()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
H A D | saved_model_test.py | 122 def _validate_inputs_tensor_info_fail(self, builder, tensor_info): argument 135 def _validate_inputs_tensor_info_accept(self, builder, tensor_info): argument 144 def _validate_outputs_tensor_info_fail(self, builder, tensor_info): argument 156 def _validate_outputs_tensor_info_accept(self, builder, tensor_info): argument
|
H A D | utils_impl.py | 148 def get_tensor_from_tensor_info(tensor_info, graph=None, import_scope=None): argument 192 def get_element_from_tensor_info(tensor_info, graph=None, import_scope=None): argument
|
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/runtime/gpu/cl/ |
H A D | ClWorkloadRuntime.cpp | 67 TensorInfo tensor_info{}; /**< Associated tensor info */ member 91 CLTensor *add_aux_tensor(const ITensorInfo &tensor_info, const AuxMemoryInfo &aux_memory_info) in add_aux_tensor() 131 const TensorInfo tensor_info = *workload_arg->tensor_info(); in create_aux_tensors() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/ |
H A D | convert_saved_model.py | 29 def _log_tensor_details(tensor_info): argument 103 def gather_names(tensor_info): argument
|
/aosp_15_r20/external/ComputeLibrary/src/core/CPP/ |
H A D | Validate.h | 42 const ITensorInfo *tensor_info) in error_on_unsupported_cpu_fp16() 65 const ITensorInfo *tensor_info) in error_on_unsupported_cpu_bf16()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/memory_management/ |
H A D | greedy_by_breadth_assignment.cc | 59 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
|
/aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/utils/ |
H A D | export.py | 87 def get_ones_tensor(tensor_info, logger): argument 95 def get_tensor_with_encoding(tensor, tensor_info, logger): argument
|
/aosp_15_r20/external/ComputeLibrary/utils/ |
H A D | ImageLoader.h | 355 virtual void validate_info(const ITensorInfo *tensor_info) in validate_info() 414 void validate_info(const ITensorInfo *tensor_info) override in validate_info()
|
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/codegen/ |
H A D | android_java_generator.cc | 110 TensorInfo tensor_info; in CreateTensorInfo() local 215 const TensorInfo& tensor_info) { in SetCodeWriterWithTensorInfo()
|
/aosp_15_r20/external/ComputeLibrary/arm_compute/dynamic_fusion/sketch/gpu/ |
H A D | GpuWorkloadSketch.h | 72 auto tensor_info = TensorInfo(std::forward<Args>(args)...); in create_tensor_info() local
|
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/ |
H A D | NENormalizationLayer.cpp | 49 TensorInfo tensor_info(input->info()->tensor_shape(), 1, input->info()->data_type()); in configure() local
|
/aosp_15_r20/external/executorch/runtime/executor/test/ |
H A D | method_meta_test.cpp | 54 void check_tensor(const TensorInfo& tensor_info) { in check_tensor()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
H A D | meta_graph.proto | 334 TensorInfo tensor_info = 1; field
|
/aosp_15_r20/external/tensorflow/tensorflow/cc/tools/ |
H A D | freeze_saved_model.cc | 37 void GetTensorNamesFromTensorInfo(const TensorInfo& tensor_info, in GetTensorNamesFromTensorInfo()
|