Searched refs:SetTensorData (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/ |
| H A D | tflite_driver.cc | 78 void SetTensorData(const std::vector<T>& values, void* data) { in SetTensorData() function
131 SetTensorData(values, data_.get()); in SetData()
594 SetTensorData(values, tensor->data.raw); in SetInput()
600 SetTensorData(values, tensor->data.raw); in SetInput()
606 SetTensorData(values, tensor->data.raw); in SetInput()
612 SetTensorData(values, tensor->data.raw); in SetInput()
618 SetTensorData(values, tensor->data.raw); in SetInput()
624 SetTensorData(values, tensor->data.raw); in SetInput()
630 SetTensorData(values, tensor->data.raw); in SetInput()
636 SetTensorData(values, tensor->data.raw); in SetInput()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| H A D | tflite_driver.cc | 60 void SetTensorData(const std::vector<T>& values, TfLitePtrUnion* data) { in SetTensorData() function
82 SetTensorData(values, &data_); in SetData()
278 SetTensorData(values, &tensor->data); in SetInput()
284 SetTensorData(values, &tensor->data); in SetInput()
290 SetTensorData(values, &tensor->data); in SetInput()
296 SetTensorData(values, &tensor->data); in SetInput()
302 SetTensorData(values, &tensor->data); in SetInput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | hashtable_ops_test.cc | 65 void SetTensorData(Interpreter* interpreter, int tensorId, in SetTensorData() function
76 void SetTensorData(Interpreter* interpreter, int tensorId, in SetTensorData() function
424 SetTensorData(interpreter_.get(), kKeyTensorId, keys_); in AllocateTensors()
425 SetTensorData(interpreter_.get(), kValueTensorId, values_); in AllocateTensors()
426 SetTensorData(interpreter_.get(), kQueryTensorId, queries_); in AllocateTensors()
427 SetTensorData(interpreter_.get(), kDefaultValueTensorId, in AllocateTensors()
431 SetTensorData(interpreter_.get(), kKeyTwoTensorId, keys_two_); in AllocateTensors()
432 SetTensorData(interpreter_.get(), kValueTwoTensorId, values_two_); in AllocateTensors()
433 SetTensorData(interpreter_.get(), kQueryTwoTensorId, queries_two_); in AllocateTensors()
434 SetTensorData(interpreter_.get(), kDefaultValueTwoTensorId, in AllocateTensors()
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | tensor.cpp | 210 void LazyTensor::SetTensorData(at::Tensor tensor_data) { in SetTensorData() function in torch::lazy::LazyTensor
248 SetTensorData(tensor); in ToTensor()
272 SetTensorData(tensor); in SetTensor()
282 SetTensorData(tensor); in UpdateFromTensor()
|
| H A D | tensor.h | 143 void SetTensorData(at::Tensor tensor_data);
|
| H A D | lazy_graph_executor.h | 361 std::vector<BackendDataPtr> SetTensorData(
|
| H A D | lazy_graph_executor.cpp | 672 std::vector<torch::lazy::BackendDataPtr> LazyGraphExecutor::SetTensorData( in SetTensorData() function in torch::lazy::LazyGraphExecutor
950 SetTensorData(tensors, coll->config, coll->indices, tensor_data_vec); in ScheduleSyncTensorsGraph()
|