Searched refs:StoreDatasetInVariantTensor (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | dataset.cc | 352 Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor) { in StoreDatasetInVariantTensor() function
390 StoreDatasetInVariantTensor(const_cast<DatasetBase*>(dataset), &t)); in AddInputDataset()
409 OP_REQUIRES_OK(ctx, StoreDatasetInVariantTensor(dataset, output)); in Compute()
|
| D | dataset.h | 584 Status StoreDatasetInVariantTensor(DatasetBase* dataset, Tensor* tensor);
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op_test.cc | 40 StoreDatasetInVariantTensor(tensor_slice_dataset, &dataset_tensor)); in CreateTensorSliceDatasetTensors()
|
| D | zip_dataset_op_test.cc | 44 StoreDatasetInVariantTensor(range_dataset, &dataset_tensor)); in CreateRangeDatasetTensors()
|
| D | window_dataset_op.cc | 244 TF_RETURN_IF_ERROR(StoreDatasetInVariantTensor(window_dataset, in GetNextInternal()
|
| D | repeat_dataset_op_test.cc | 35 StoreDatasetInVariantTensor(tensor_slice_dataset, dataset_tensor)); in CreateTensorSliceDatasetTensor()
|
| D | map_dataset_op_test.cc | 73 StoreDatasetInVariantTensor(input_dataset, &dataset_tensor)); in CreateMapDatasetContext()
|
| D | take_dataset_op_test.cc | 32 StoreDatasetInVariantTensor(tensor_slice_dataset, dataset_tensor)); in CreateTensorSliceDatasetTensor()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | choose_fastest_branch_dataset_op.cc | 505 TF_RETURN_IF_ERROR(StoreDatasetInVariantTensor( in MakeCurrentIterator()
|
| D | group_by_window_dataset_op.cc | 452 StoreDatasetInVariantTensor(group_dataset, &group_dataset_arg)); in StartFlushingGroup()
|