Home
last modified time | relevance | path

Searched defs:tensor_ptr (Results 1 – 19 of 19) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/
H A Dinternal_api_samples.cc39 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 Ddelegate_testing.cc35 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 Dxnnpack_graph_builder.cpp154 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 Dslice_op.cc44 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 Dquantized_add_op.cc352 const uint8* tensor_ptr = &(tensor_data->value) + i; in VectorTensorAddition() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter.h278 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 Devalue_test.cpp234 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 Dmemory_planner.cpp85 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 Dmemory_planner.h208 uint8_t* tensor_ptr = in isManagedOutputTensor() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/core/
H A Dtask_utils.h41 T* TypedTensor(const TfLiteTensor* tensor_ptr) { in TypedTensor()
/aosp_15_r20/external/executorch/extension/tensor/
H A Dtensor_ptr.cpp107 const auto tensor_ptr = &storage->tensor; in make_tensor_ptr() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dembedding_lookup_sparse_test.cc71 float* tensor_ptr = GetTensorData<float>(tensor); in Set3DWeightMatrix() local
H A Dtest_util.cc320 TfLiteTensor* tensor_ptr = interpreter_->tensor(index); in ExtractVector() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
H A Dparse_testdata.cc101 &csv](FloatTensor* tensor_ptr) { in ParseExamples()
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/
H A Dparse_testdata.cc103 &csv](FloatTensor* tensor_ptr) { in ParseExamples()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dinference_context.cc772 MetalSpatialTensor* tensor_ptr) { in SetTensor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
H A Dtrt_engine_op.cc283 void* GetTensorAddress(const Tensor* tensor_ptr) { in GetTensorAddress()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dinference_context.cc700 Tensor* tensor_ptr) { in SetTensor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc2275 Tensor* tensor_ptr = context->mutable_output(0); in TestConvertVariableV2() local