| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| H A D | internal_api_samples.cc | 39 TfLiteTensor* tensor_ptr = interpreter->tensor(interpreter->inputs()[k]); in FillInputTensors() local
40 const auto tensor_elements_count = tflite::NumElements(tensor_ptr); in FillInputTensors()
41 if (tensor_ptr->type == kTfLiteFloat32) { in FillInputTensors()
58 TfLiteTensor* tensor_ptr = cpu->tensor(outputs[i]); in CompareCPUGPUResults() local
59 const float* cpu_out = tensor_ptr->data.f; in CompareCPUGPUResults()
64 for (int k = 0; k < tensor_ptr->bytes / 4; ++k) { in CompareCPUGPUResults()
115 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPI() local
117 if (tensor_ptr->type != kTfLiteFloat32) { in RunModelSampleWithInternalAPI()
123 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPI() local
125 if (tensor_ptr->type != kTfLiteFloat32) { in RunModelSampleWithInternalAPI()
[all …]
|
| H A D | delegate_testing.cc | 35 TfLiteTensor* tensor_ptr = interpreter->tensor(interpreter->inputs()[k]); in FillInputTensor() local
36 const auto tensor_elements_count = tflite::NumElements(tensor_ptr); in FillInputTensor()
37 if (tensor_ptr->type == kTfLiteFloat32) { in FillInputTensor()
43 if (tensor_ptr->type == kTfLiteInt32) { in FillInputTensor()
49 if (tensor_ptr->type == kTfLiteInt8) { in FillInputTensor()
55 if (tensor_ptr->type == kTfLiteUInt8) { in FillInputTensor()
67 TfLiteTensor* tensor_ptr = cpu->tensor(cpu->outputs()[i]); in CompareCPUGPUResults() local
68 const auto tensor_elements_count = tflite::NumElements(tensor_ptr); in CompareCPUGPUResults()
70 std::cout << "Output " << tensor_ptr->name << ":" << std::endl; in CompareCPUGPUResults()
78 if (tensor_ptr->type == kTfLiteFloat32) { in CompareCPUGPUResults()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/benchmarking/ |
| H A D | main.mm | 45 TfLiteTensor* tensor_ptr = interpreter->tensor(interpreter->inputs()[k]);
46 const auto tensor_elements_count = tflite::NumElements(tensor_ptr);
47 if (tensor_ptr->type == kTfLiteFloat32) {
53 if (tensor_ptr->type == kTfLiteInt32) {
59 if (tensor_ptr->type == kTfLiteInt8) {
65 if (tensor_ptr->type == kTfLiteUInt8) {
77 TfLiteTensor* tensor_ptr = cpu->tensor(outputs[i]->tensor.ref);
78 const auto tensor_elements_count = tflite::NumElements(tensor_ptr);
80 std::cout << "Output " << tensor_ptr->name << ":" << std::endl;
91 if (tensor_ptr->type == kTfLiteFloat32) {
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/ |
| H A D | xnnpack_graph_builder.cpp | 154 auto tensor_ptr = val->type()->cast<TensorType>(); in getGraphOutputShapes() local
155 std::vector<long> sizes = tensor_ptr->sizes().concrete_sizes().value(); in getGraphOutputShapes()
212 auto tensor_ptr = val->type()->cast<TensorType>(); in defineAllTensorValues() local
213 auto num_dims = tensor_ptr->dim().value(); in defineAllTensorValues()
217 std::vector<long> sizes = tensor_ptr->sizes().concrete_sizes().value(); in defineAllTensorValues()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | interpreter.h | 278 if (TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
279 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_tensor()
280 return reinterpret_cast<T*>(tensor_ptr->data.raw); in typed_tensor()
290 if (const TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
291 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_tensor()
292 return reinterpret_cast<const T*>(tensor_ptr->data.raw); in typed_tensor()
|
| /aosp_15_r20/external/executorch/runtime/core/test/ |
| H A D | evalue_test.cpp | 234 auto tensor_ptr = std::make_unique<exec_aten::Tensor>(tf.ones({2, 3})); in TEST_F() local
236 EValue evalue(std::move(tensor_ptr)); in TEST_F()
251 auto tensor_ptr = std::make_shared<exec_aten::Tensor>(tf.ones({4, 5})); in TEST_F() local
253 EValue evalue(tensor_ptr); in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | slice_op.cc | 44 const int32* tensor_ptr = tensor.flat<int32>().data(); in IntTensorToInt64Vec() local
46 out_ptr[i] = tensor_ptr[i]; in IntTensorToInt64Vec()
49 const int64_t* tensor_ptr = tensor.flat<int64_t>().data(); in IntTensorToInt64Vec() local
51 out_ptr[i] = tensor_ptr[i]; in IntTensorToInt64Vec()
|
| H A D | quantized_add_op.cc | 352 const uint8* tensor_ptr = &(tensor_data->value) + i; in VectorTensorAddition() local
354 Requantize8x8To32Neon(tensor_ptr, tensor_0_64x2, tensor_mult_32x2); in VectorTensorAddition()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| H A D | task_utils.h | 41 T* TypedTensor(const TfLiteTensor* tensor_ptr) { in TypedTensor() argument
42 if (tensor_ptr->type == typeToTfLiteType<T>()) { in TypedTensor()
43 return reinterpret_cast<T*>(tensor_ptr->data.raw); in TypedTensor()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | memory_planner.h | 208 uint8_t* tensor_ptr = in isManagedOutputTensor() local
212 return buffer_start <= tensor_ptr && tensor_ptr < buffer_end; in isManagedOutputTensor()
|
| H A D | memory_planner.cpp | 85 auto* tensor_ptr = tensor_value_to_tensor.at(value); in assignStorageToManagedTensors() local
86 managed_tensor_groups.emplace_back(tensor_ptr); in assignStorageToManagedTensors()
94 auto* tensor_ptr = tensor_value_to_tensor.at(value); in assignStorageToManagedTensors() local
95 managed_tensor_groups[storage_group].addTensor(tensor_ptr); in assignStorageToManagedTensors()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | test_util.cc | 320 TfLiteTensor* tensor_ptr = interpreter_->tensor(index); in ExtractVector() local
321 CHECK(tensor_ptr != nullptr); in ExtractVector()
322 const int num_strings = GetStringCount(tensor_ptr); in ExtractVector()
326 const auto str = GetString(tensor_ptr, i); in ExtractVector()
|
| H A D | embedding_lookup_sparse_test.cc | 71 float* tensor_ptr = GetTensorData<float>(tensor); in Set3DWeightMatrix() local
75 tensor_ptr[(i * columns + j) * features + k] = function(i, j, k); in Set3DWeightMatrix()
|
| /aosp_15_r20/external/executorch/extension/tensor/ |
| H A D | targets.bzl | 16 "tensor_ptr.cpp",
21 "tensor_ptr.h",
|
| H A D | tensor_ptr.cpp | 107 const auto tensor_ptr = &storage->tensor; in make_tensor_ptr() local
108 return std::shared_ptr<exec_aten::Tensor>(std::move(storage), tensor_ptr); in make_tensor_ptr()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_op.cc | 283 void* GetTensorAddress(const Tensor* tensor_ptr) { in GetTensorAddress() argument
284 auto tensor_type = tensor_ptr->dtype(); in GetTensorAddress()
286 TYPECASE(DT_FLOAT, tensor_ptr, dest_ptr); in GetTensorAddress()
287 TYPECASE(DT_HALF, tensor_ptr, dest_ptr); in GetTensorAddress()
288 TYPECASE(DT_INT8, tensor_ptr, dest_ptr); in GetTensorAddress()
289 TYPECASE(DT_INT32, tensor_ptr, dest_ptr); in GetTensorAddress()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| H A D | inference_context.cc | 772 MetalSpatialTensor* tensor_ptr) { in SetTensor() argument
777 external_mutable_tensors_[tensor_id] = tensor_ptr; in SetTensor()
782 node.task.SetSrcTensor(tensor_ptr, i); in SetTensor()
787 node.task.SetDstTensor(tensor_ptr, i); in SetTensor()
|
| H A D | inference_context.h | 129 MetalSpatialTensor* tensor_ptr);
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| H A D | parse_testdata.cc | 101 &csv](FloatTensor* tensor_ptr) { in ParseExamples() argument
107 FloatTensor& tensor = *tensor_ptr; in ParseExamples()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/ |
| H A D | parse_testdata.cc | 103 &csv](FloatTensor* tensor_ptr) { in ParseExamples() argument
109 FloatTensor& tensor = *tensor_ptr; in ParseExamples()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| H A D | inference_context.cc | 700 Tensor* tensor_ptr) { in SetTensor() argument
705 external_mutable_tensors_[tensor_id] = tensor_ptr; in SetTensor()
710 RETURN_IF_ERROR(node.cl_operation.SetSrcTensor(i, tensor_ptr)); in SetTensor()
715 RETURN_IF_ERROR(node.cl_operation.SetDstTensor(i, tensor_ptr)); in SetTensor()
|
| H A D | inference_context.h | 111 absl::Status SetTensor(const ValueId& tensor_id, Tensor* tensor_ptr);
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/cl_kernels/common/ |
| H A D | memset.cl | 35 …* @param[in] tensor_ptr Pointer to the source image. Data types supporte…
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes_test.cc | 2275 Tensor* tensor_ptr = context->mutable_output(0); in TestConvertVariableV2() local
2276 CHECK_NOTNULL(tensor_ptr); in TestConvertVariableV2()
2284 context->allocate_temp(dtype, shape, tensor_ptr, attr)); in TestConvertVariableV2()
2286 auto tensor_flat = tensor_ptr->flat<CType>(); in TestConvertVariableV2()
|