Searched refs:StoreDatasetInVariantTensor (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.cc | 413 Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor) { in StoreDatasetInVariantTensor() function
473 StoreDatasetInVariantTensor(const_cast<DatasetBase*>(dataset), &t)); in AddInputDataset()
655 OP_REQUIRES_OK(ctx, StoreDatasetInVariantTensor(dataset, output)); in Compute()
|
| D | dataset.h | 800 Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor);
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | io_ops.cc | 311 TF_RETURN_IF_ERROR(StoreDatasetInVariantTensor(dataset_of_snapshot_files, in InitializeInput()
|
| D | group_by_window_dataset_op.cc | 485 StoreDatasetInVariantTensor(group_dataset, &group_dataset_arg)); in StartFlushingGroup()
|
| D | choose_fastest_branch_dataset_op.cc | 532 TF_RETURN_IF_ERROR(StoreDatasetInVariantTensor( in MakeCurrentIterator()
|
| D | snapshot_util.cc | 545 StoreDatasetInVariantTensor(dataset()->datasets_[index_], &tensor)); in GetNextInternal()
|
| D | snapshot_dataset_op.cc | 585 TF_RETURN_IF_ERROR(StoreDatasetInVariantTensor(dataset_of_snapshot_files, in Initialize()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | window_dataset_op.cc | 254 StoreDatasetInVariantTensor(window_dataset, &out_tensors->back())); in GetNextInternal()
|
| D | dataset_test_base.cc | 946 StoreDatasetInVariantTensor(dataset_base, &dataset_tensor)); in MakeDatasetTensor()
|