Home
last modified time | relevance | path

Searched refs:tensor_slice_dataset (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op_test.cc158 DatasetBase *tensor_slice_dataset; in TEST_F() local
161 &tensor_slice_dataset)); in TEST_F()
162 core::ScopedUnref scored_unref(tensor_slice_dataset); in TEST_F()
168 TF_ASSERT_OK(tensor_slice_dataset->MakeIterator(iterator_context.get(), in TEST_F()
220 DatasetBase *tensor_slice_dataset; in TEST_F() local
223 &tensor_slice_dataset)); in TEST_F()
224 core::ScopedUnref scored_unref(tensor_slice_dataset); in TEST_F()
226 EXPECT_EQ(tensor_slice_dataset->type_string(), kOpName); in TEST_F()
257 DatasetBase *tensor_slice_dataset; in TEST_F() local
260 &tensor_slice_dataset)); in TEST_F()
[all …]
Dconcatenate_dataset_op_test.cc34 DatasetBase *tensor_slice_dataset; in CreateTensorSliceDatasetTensors() local
37 &tensors, &tensor_slice_dataset)); in CreateTensorSliceDatasetTensors()
40 StoreDatasetInVariantTensor(tensor_slice_dataset, &dataset_tensor)); in CreateTensorSliceDatasetTensors()
Ddataset_test_base.cc58 DatasetBase** tensor_slice_dataset) { in CreateTensorSliceDataset() argument
85 GetDatasetFromContext(context.get(), 0, tensor_slice_dataset)); in CreateTensorSliceDataset()
Dtake_dataset_op_test.cc28 DatasetBase *tensor_slice_dataset; in CreateTensorSliceDatasetTensor() local
30 "tensor_slice_node", tensor_vector, &tensor_slice_dataset)); in CreateTensorSliceDatasetTensor()
32 StoreDatasetInVariantTensor(tensor_slice_dataset, dataset_tensor)); in CreateTensorSliceDatasetTensor()
Drepeat_dataset_op_test.cc31 DatasetBase *tensor_slice_dataset; in CreateTensorSliceDatasetTensor() local
33 "tensor_slice_node", tensor_vector, &tensor_slice_dataset)); in CreateTensorSliceDatasetTensor()
35 StoreDatasetInVariantTensor(tensor_slice_dataset, dataset_tensor)); in CreateTensorSliceDatasetTensor()
Ddataset_test_base.h120 DatasetBase** tensor_slice_dataset);
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py2111 variant_tensor = gen_dataset_ops.tensor_slice_dataset(