Searched refs:tensors_data (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/csrc/lazy/core/ |
| D | lazy_graph_executor.cpp | 75 devctx->tensors_data.emplace(data->unique_id, data); in RegisterTensor()
82 devctx->tensors_data.erase(data->unique_id); in UnregisterTensor()
90 for (auto& uid_wptr : devctx->tensors_data) { in GetLiveTensors()
539 std::vector<BackendDataPtr> tensors_data, in Async() argument
547 tensors_data(std::move(tensors_data)) {} in Async()
677 std::vector<BackendDataPtr> tensors_data; in SetTensorData() local
678 tensors_data.reserve(indices.size()); in SetTensorData()
700 tensors_data.emplace_back(std::move(handle)); in SetTensorData()
702 return tensors_data; in SetTensorData()
883 std::vector<BackendDataPtr> tensors_data, in ScheduleSyncTensorsGraph() argument
[all …]
|
| D | lazy_graph_executor.h | 270 std::map<int64_t, std::weak_ptr<LazyTensor::Data>> tensors_data; member
318 std::vector<BackendDataPtr> tensors_data,
330 std::vector<BackendDataPtr> tensors_data; member
400 std::vector<BackendDataPtr> tensors_data,
414 c10::ArrayRef<BackendDataPtr> tensors_data,
422 c10::ArrayRef<BackendDataPtr> tensors_data);
|
| /external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/ |
| D | nnapi_plugin_test.cc | 235 const tflite::TensorData tensors_data = {tflite::TensorType_FLOAT32, in Build() local
237 int input1 = AddInput(tensors_data); in Build()
238 int input2 = AddInput(tensors_data); in Build()
239 int input3 = AddInput(tensors_data); in Build()
240 int add_out = AddInnerTensor<float>(tensors_data); in Build()
241 int round_out = AddInnerTensor<float>(tensors_data); in Build()
242 int output = AddOutput(tensors_data); in Build()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 314 std::vector<const char*> tensors_data; in Compute() local
330 tensors_data.push_back( in Compute()
335 shape_.IsTuple() ? xla::BorrowingLiteral(tensors_data, shape_) in Compute()
336 : xla::BorrowingLiteral(tensors_data.front(), shape_); in Compute()
|