Searched refs:tensor_slice_dataset (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | tensor_slice_dataset_op_test.cc | 158 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 …]
|
D | concatenate_dataset_op_test.cc | 34 DatasetBase *tensor_slice_dataset; in CreateTensorSliceDatasetTensors() local 37 &tensors, &tensor_slice_dataset)); in CreateTensorSliceDatasetTensors() 40 StoreDatasetInVariantTensor(tensor_slice_dataset, &dataset_tensor)); in CreateTensorSliceDatasetTensors()
|
D | dataset_test_base.cc | 58 DatasetBase** tensor_slice_dataset) { in CreateTensorSliceDataset() argument 85 GetDatasetFromContext(context.get(), 0, tensor_slice_dataset)); in CreateTensorSliceDataset()
|
D | take_dataset_op_test.cc | 28 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()
|
D | repeat_dataset_op_test.cc | 31 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()
|
D | dataset_test_base.h | 120 DatasetBase** tensor_slice_dataset);
|
/external/tensorflow/tensorflow/python/data/ops/ |
D | dataset_ops.py | 2111 variant_tensor = gen_dataset_ops.tensor_slice_dataset(
|