Searched refs:StoreDatasetInVariantTensor (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.cc | 475 Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor) { in StoreDatasetInVariantTensor() function
765 StoreDatasetInVariantTensor(const_cast<DatasetBase*>(dataset), &t)); in AddInputDataset()
1002 OP_REQUIRES_OK(ctx, StoreDatasetInVariantTensor(dataset, output)); in Compute()
|
| D | dataset.h | 889 Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | window_dataset_op.cc | 253 StoreDatasetInVariantTensor(window_dataset, &out_tensors->back())); in GetNextInternal()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | group_by_window_dataset_op.cc | 488 StoreDatasetInVariantTensor(group_dataset, &group_dataset_arg)); in StartFlushingGroup()
|
| D | choose_fastest_branch_dataset_op.cc | 532 TF_RETURN_IF_ERROR(StoreDatasetInVariantTensor( in MakeCurrentIterator()
|
| D | io_ops.cc | 699 TF_RETURN_IF_ERROR(StoreDatasetInVariantTensor(dataset_of_snapshot_files, in InitializeInput()
|
| D | snapshot_dataset_op.cc | 293 TF_RETURN_IF_ERROR(StoreDatasetInVariantTensor(dataset_of_snapshot_files, in Initialize()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | snapshot_utils.cc | 621 StoreDatasetInVariantTensor(dataset()->datasets_[index_], &tensor)); in GetNextInternal()
|
| D | dataset_test_base.cc | 978 StoreDatasetInVariantTensor(dataset_base, &dataset_tensor)); in MakeDatasetTensor()
|