Searched refs:created_tensors (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 798 std::vector<std::unique_ptr<Tensor>> created_tensors; in MakeDataset() local
801 &created_tensors, &dataset_base)); in MakeDataset()
804 std::move(dataset_ctx), std::move(created_tensors), dataset_base); in MakeDataset()
812 std::vector<std::unique_ptr<Tensor>>* created_tensors, in RunDatasetOp() argument
817 TF_RETURN_IF_ERROR(MakeDatasetTensor(*input, created_tensors, &t)); in RunDatasetOp()
819 created_tensors->push_back(std::move(t)); in RunDatasetOp()
831 created_tensors->push_back(std::move(copy)); in RunDatasetOp()
846 std::vector<std::unique_ptr<Tensor>>* created_tensors, in MakeDataset() argument
849 dataset_ctx_params, created_tensors, in MakeDataset()
912 std::vector<std::unique_ptr<Tensor>>* created_tensors, in MakeDatasetTensor() argument
[all …]
|
| D | dataset_test_base.h | 725 std::vector<std::unique_ptr<Tensor>>* created_tensors,
733 std::vector<std::unique_ptr<Tensor>>* created_tensors,
743 std::vector<std::unique_ptr<Tensor>>* created_tensors,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 523 std::vector<bool> created_tensors(buffer_usage_records.size(), false); in AllocateMemoryForBuffers() local
532 if (created_tensors[tensor_index]) continue; in AllocateMemoryForBuffers()
538 created_tensors[tensor_index] = true; in AllocateMemoryForBuffers()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 555 std::vector<bool> created_tensors(buffer_usage_records.size(), false); in AllocateMemoryForBuffers() local
562 if (created_tensors[tensor_index]) continue; in AllocateMemoryForBuffers()
568 created_tensors[tensor_index] = true; in AllocateMemoryForBuffers()
|