Searched refs:range_dataset (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | range_dataset_op_test.cc | 85 DatasetBase* range_dataset; in TEST_P() local
87 CreateDataset(range_kernel.get(), range_context.get(), &range_dataset)); in TEST_P()
88 core::ScopedUnref scoped_unref(range_dataset); in TEST_P()
93 TF_ASSERT_OK(range_dataset->MakeIterator(iterator_context.get(), "Iterator", in TEST_P()
134 DatasetBase* range_dataset; in TEST_F() local
136 CreateDataset(range_kernel.get(), range_context.get(), &range_dataset)); in TEST_F()
137 core::ScopedUnref scoped_unref(range_dataset); in TEST_F()
139 EXPECT_EQ(range_dataset->type_string(), kOpName); in TEST_F()
154 DatasetBase* range_dataset; in TEST_F() local
156 CreateDataset(range_kernel.get(), range_context.get(), &range_dataset)); in TEST_F()
[all …]
|
| D | map_dataset_op_test.cc | 119 DatasetBase* range_dataset; in TEST_P() local
122 &range_dataset)); in TEST_P()
123 core::ScopedUnref scored_unref_range_dataset(range_dataset); in TEST_P()
127 range_dataset->node_name(), test_params.func_name, &map_kernel)); in TEST_P()
130 CreateMapDatasetContext(range_dataset, map_kernel.get(), &map_context)); in TEST_P()
177 DatasetBase* range_dataset; in TEST_F() local
179 CreateRangeDataset<int64>(start, end, step, "range", &range_dataset)); in TEST_F()
180 core::ScopedUnref scoped_unref_range_dataset(range_dataset); in TEST_F()
184 range_dataset->node_name(), func_def.signature().name(), &map_kernel)); in TEST_F()
187 CreateMapDatasetContext(range_dataset, map_kernel.get(), &map_context)); in TEST_F()
[all …]
|
| D | dataset_test_base.h | 91 DatasetBase** range_dataset) { in CreateRangeDataset() argument
107 GetDatasetFromContext(range_context.get(), 0, range_dataset)); in CreateRangeDataset()
|
| D | zip_dataset_op_test.cc | 38 DatasetBase *range_dataset; in CreateRangeDatasetTensors() local
41 strings::StrCat("range_", i), &range_dataset)); in CreateRangeDatasetTensors()
44 StoreDatasetInVariantTensor(range_dataset, &dataset_tensor)); in CreateRangeDatasetTensors()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization_test.cc | 245 auto range_dataset = AddRangeNode(&graph); in TEST_P() local
247 auto dataset = AddMapNode(&graph, range_dataset->name(), in TEST_P()
276 range_dataset->name()); in TEST_P()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 2619 variant_tensor = gen_dataset_ops.range_dataset(
|