Searched refs:TakeDatasetOp (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | take_dataset_op.cc | 24 /* static */ constexpr const char* const TakeDatasetOp::kDatasetType;
25 /* static */ constexpr const char* const TakeDatasetOp::kInputDataset;
26 /* static */ constexpr const char* const TakeDatasetOp::kCount;
27 /* static */ constexpr const char* const TakeDatasetOp::kOutputTypes;
28 /* static */ constexpr const char* const TakeDatasetOp::kOutputShapes;
60 return name_utils::DatasetDebugString(TakeDatasetOp::kDatasetType); in DebugString()
206 TakeDatasetOp::TakeDatasetOp(OpKernelConstruction* ctx) in TakeDatasetOp() function in tensorflow::data::TakeDatasetOp
209 void TakeDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()
218 REGISTER_KERNEL_BUILDER(Name("TakeDataset").Device(DEVICE_CPU), TakeDatasetOp);
|
| D | take_dataset_op.h | 59 class TakeDatasetOp : public UnaryDatasetOpKernel {
67 explicit TakeDatasetOp(OpKernelConstruction* ctx);
|
| D | take_dataset_op_test.cc | 91 CheckDatasetTypeString(name_utils::OpName(TakeDatasetOp::kDatasetType))); in TEST_F()
|
| D | dataset_test_base.cc | 1126 *input_names = {TakeDatasetOp::kInputDataset, TakeDatasetOp::kCount}; in GetInputNames()
1131 *attr_vector = {{TakeDatasetOp::kOutputShapes, output_shapes_}, in GetAttributes()
1132 {TakeDatasetOp::kOutputTypes, output_dtypes_}}; in GetAttributes()
1137 return TakeDatasetOp::kDatasetType; in dataset_type()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | assert_next_dataset_op_test.cc | 80 /*transformations=*/{TakeDatasetOp::kDatasetType}, in AssertNextDatasetParams1()
96 {TakeDatasetOp::kDatasetType, RangeDatasetOp::kDatasetType}, in AssertNextDatasetParams2()
126 {TakeDatasetOp::kDatasetType, RangeDatasetOp::kDatasetType, "Whoops"}, in ShortAssertNextDatasetParams()
|