Home
last modified time | relevance | path

Searched refs:SetTensorData (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/
H A Dtflite_driver.cc78 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 Dtflite_driver.cc60 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 Dhashtable_ops_test.cc65 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 Dtensor.cpp210 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 Dtensor.h143 void SetTensorData(at::Tensor tensor_data);
H A Dlazy_graph_executor.h361 std::vector<BackendDataPtr> SetTensorData(
H A Dlazy_graph_executor.cpp672 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()