/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 58 TfLiteTensor* tensor_ptr = cpu->tensor(outputs[i]); in CompareCPUGPUResults() local 115 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPI() local 123 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPI() local 201 TfLiteTensor* tensor_ptr = cpu_inference->tensor(in_refs[i]); in RunModelSampleWithInternalAPI() local 208 TfLiteTensor* tensor_ptr = cpu_inference->tensor(out_refs[i]); in RunModelSampleWithInternalAPI() local 239 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPISerializedKernels() local 247 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPISerializedKernels() local 326 TfLiteTensor* tensor_ptr = cpu_inference->tensor(in_refs[i]); in RunModelSampleWithInternalAPISerializedKernels() local 333 TfLiteTensor* tensor_ptr = cpu_inference->tensor(out_refs[i]); in RunModelSampleWithInternalAPISerializedKernels() local [all …]
|
H A D | delegate_testing.cc | 35 TfLiteTensor* tensor_ptr = interpreter->tensor(interpreter->inputs()[k]); in FillInputTensor() local 67 TfLiteTensor* tensor_ptr = cpu->tensor(cpu->outputs()[i]); in CompareCPUGPUResults() local
|
/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 212 auto tensor_ptr = val->type()->cast<TensorType>(); in defineAllTensorValues() local
|
/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 49 const int64_t* tensor_ptr = tensor.flat<int64_t>().data(); in IntTensorToInt64Vec() local
|
H A D | quantized_add_op.cc | 352 const uint8* tensor_ptr = &(tensor_data->value) + i; in VectorTensorAddition() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/ |
H A D | interpreter.h | 278 if (TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local 290 if (const TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
|
/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 251 auto tensor_ptr = std::make_shared<exec_aten::Tensor>(tf.ones({4, 5})); in TEST_F() local
|
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
H A D | memory_planner.cpp | 85 auto* tensor_ptr = tensor_value_to_tensor.at(value); in assignStorageToManagedTensors() local 94 auto* tensor_ptr = tensor_value_to_tensor.at(value); in assignStorageToManagedTensors() local
|
H A D | memory_planner.h | 208 uint8_t* tensor_ptr = in isManagedOutputTensor() local
|
/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()
|
/aosp_15_r20/external/executorch/extension/tensor/ |
H A D | tensor_ptr.cpp | 107 const auto tensor_ptr = &storage->tensor; in make_tensor_ptr() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | embedding_lookup_sparse_test.cc | 71 float* tensor_ptr = GetTensorData<float>(tensor); in Set3DWeightMatrix() local
|
H A D | test_util.cc | 320 TfLiteTensor* tensor_ptr = interpreter_->tensor(index); in ExtractVector() local
|
/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()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/ |
H A D | parse_testdata.cc | 103 &csv](FloatTensor* tensor_ptr) { in ParseExamples()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
H A D | inference_context.cc | 772 MetalSpatialTensor* tensor_ptr) { in SetTensor()
|
/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()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
H A D | inference_context.cc | 700 Tensor* tensor_ptr) { in SetTensor()
|
/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
|