Searched refs:RangeDatasetOp (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | range_dataset_op.cc | 29 /* static */ constexpr const char* const RangeDatasetOp::kDatasetType;
30 /* static */ constexpr const char* const RangeDatasetOp::kStart;
31 /* static */ constexpr const char* const RangeDatasetOp::kStop;
32 /* static */ constexpr const char* const RangeDatasetOp::kStep;
33 /* static */ constexpr const char* const RangeDatasetOp::kOutputTypes;
34 /* static */ constexpr const char* const RangeDatasetOp::kOutputShapes;
89 class RangeDatasetOp::RangeSplitProvider : public SplitProvider {
128 class RangeDatasetOp::Dataset : public DatasetBase {
299 RangeDatasetOp::RangeDatasetOp(OpKernelConstruction* ctx) in RangeDatasetOp() function in tensorflow::data::RangeDatasetOp
304 void RangeDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase** output) { in MakeDataset()
[all …]
|
| D | range_dataset_op.h | 23 class RangeDatasetOp : public DatasetOpKernel {
32 explicit RangeDatasetOp(OpKernelConstruction* ctx);
|
| D | name_utils_test.cc | 38 EXPECT_EQ(name_utils::DatasetDebugString(RangeDatasetOp::kDatasetType, in TEST()
58 EXPECT_EQ(name_utils::OpName(RangeDatasetOp::kDatasetType), "RangeDataset"); in TEST()
|
| D | range_dataset_op_test.cc | 69 CheckDatasetTypeString(name_utils::OpName(RangeDatasetOp::kDatasetType))); in TEST_F()
120 RangeDatasetOp::kDatasetType, range_dataset_params.iterator_prefix()))); in TEST_F()
|
| D | dataset_test_base.cc | 995 *input_names = {RangeDatasetOp::kStart, RangeDatasetOp::kStop, in GetInputNames()
996 RangeDatasetOp::kStep}; in GetInputNames()
1001 *attr_vector = {{RangeDatasetOp::kOutputTypes, output_dtypes_}, in GetAttributes()
1002 {RangeDatasetOp::kOutputShapes, output_shapes_}}; in GetAttributes()
1007 return RangeDatasetOp::kDatasetType; in dataset_type()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | assert_next_dataset_op_test.cc | 96 {TakeDatasetOp::kDatasetType, RangeDatasetOp::kDatasetType}, in AssertNextDatasetParams2()
126 {TakeDatasetOp::kDatasetType, RangeDatasetOp::kDatasetType, "Whoops"}, in ShortAssertNextDatasetParams()
|