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()
|