Searched refs:created_tensors (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset_test_base.cc | 813 std::vector<std::unique_ptr<Tensor>> created_tensors; in MakeDataset() local
816 &created_tensors, &dataset_base)); in MakeDataset()
819 std::move(dataset_ctx), std::move(created_tensors), dataset_base); in MakeDataset()
827 std::vector<std::unique_ptr<Tensor>>* created_tensors, in RunDatasetOp() argument
832 TF_RETURN_IF_ERROR(MakeDatasetTensor(*input, created_tensors, &t)); in RunDatasetOp()
834 created_tensors->push_back(std::move(t)); in RunDatasetOp()
846 created_tensors->push_back(std::move(copy)); in RunDatasetOp()
861 std::vector<std::unique_ptr<Tensor>>* created_tensors, in MakeDataset() argument
864 dataset_ctx_params, created_tensors, in MakeDataset()
943 std::vector<std::unique_ptr<Tensor>>* created_tensors, in MakeDatasetTensor() argument
[all …]
|
| D | dataset_test_base.h | 778 std::vector<std::unique_ptr<Tensor>>* created_tensors,
786 std::vector<std::unique_ptr<Tensor>>* created_tensors,
796 std::vector<std::unique_ptr<Tensor>>* created_tensors,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 556 std::vector<bool> created_tensors(buffer_usage_records.size(), false); in AllocateMemoryForBuffers() local
566 if (created_tensors[tensor_index]) continue; in AllocateMemoryForBuffers()
592 created_tensors[tensor_index] = true; in AllocateMemoryForBuffers()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 554 std::vector<bool> created_tensors(buffer_usage_records.size(), false); in AllocateBufferBasedTensors() local
570 if (created_tensors[tensor_index]) continue; in AllocateBufferBasedTensors()
596 created_tensors[tensor_index] = true; in AllocateBufferBasedTensors()
|