Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/utils/
H A DUtils.h38 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 DSubTensorInfo.cpp78 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 DTensorInfo.cpp190 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 Dvalidation_graph_builder.cc76 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 DPReluLayer.cpp98 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 DGpuKernelArgument.h80 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 DGpuWorkloadSourceCode.h55 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 DGpuWorkloadSketch.cpp46 void GpuWorkloadSketch::register_new_tensor(ITensorInfo &tensor_info) in register_new_tensor()
55 TensorInfo tensor_info{}; in create_tensor_info() local
H A DGpuWorkloadSketchImpl.h129 ITensorInfo *create_auxiliary_tensor(const ITensorInfo &tensor_info) in create_auxiliary_tensor()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/
H A Dsaved_model.cc402 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 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()
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dsaved_model_test.py122 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 Dutils_impl.py148 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 DClWorkloadRuntime.cpp67 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 Dconvert_saved_model.py29 def _log_tensor_details(tensor_info): argument
103 def gather_names(tensor_info): argument
/aosp_15_r20/external/ComputeLibrary/src/core/CPP/
H A DValidate.h42 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 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
/aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/utils/
H A Dexport.py87 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 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()
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/codegen/
H A Dandroid_java_generator.cc110 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 DGpuWorkloadSketch.h72 auto tensor_info = TensorInfo(std::forward<Args>(args)...); in create_tensor_info() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNENormalizationLayer.cpp49 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 Dmethod_meta_test.cpp54 void check_tensor(const TensorInfo& tensor_info) { in check_tensor()
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dmeta_graph.proto334 TensorInfo tensor_info = 1; field
/aosp_15_r20/external/tensorflow/tensorflow/cc/tools/
H A Dfreeze_saved_model.cc37 void GetTensorNamesFromTensorInfo(const TensorInfo& tensor_info, in GetTensorNamesFromTensorInfo()

12