Searched refs:ConcatenateDatasetOp (Results 1 – 5 of 5) sorted by relevance
27 /* static */ constexpr const char* const ConcatenateDatasetOp::kDatasetType;28 /* static */ constexpr const char* const ConcatenateDatasetOp::kInputDataset;29 /* static */ constexpr const char* const ConcatenateDatasetOp::kAnotherDataset;30 /* static */ constexpr const char* const ConcatenateDatasetOp::kOutputTypes;31 /* static */ constexpr const char* const ConcatenateDatasetOp::kOutputShapes;36 class ConcatenateDatasetOp::Dataset : public DatasetBase {217 ConcatenateDatasetOp::ConcatenateDatasetOp(OpKernelConstruction* ctx) in ConcatenateDatasetOp() function in tensorflow::data::ConcatenateDatasetOp220 void ConcatenateDatasetOp::MakeDataset(OpKernelContext* ctx, DatasetBase* input, in MakeDataset()234 ConcatenateDatasetOp);
23 class ConcatenateDatasetOp : public BinaryDatasetOpKernel {31 explicit ConcatenateDatasetOp(OpKernelConstruction* ctx);
34 EXPECT_EQ(name_utils::DatasetDebugString(ConcatenateDatasetOp::kDatasetType), in TEST()59 EXPECT_EQ(name_utils::OpName(ConcatenateDatasetOp::kDatasetType, in TEST()
124 name_utils::OpName(ConcatenateDatasetOp::kDatasetType))); in TEST_F()197 ConcatenateDatasetOp::kDatasetType, dataset_params.iterator_prefix()))); in ITERATOR_OUTPUT_SHAPES_TEST_P()
1146 *input_names = {ConcatenateDatasetOp::kInputDataset, in GetInputNames()1147 ConcatenateDatasetOp::kAnotherDataset}; in GetInputNames()1153 *attr_vector = {{ConcatenateDatasetOp::kOutputTypes, output_dtypes_}, in GetAttributes()1154 {ConcatenateDatasetOp::kOutputShapes, output_shapes_}}; in GetAttributes()1159 return ConcatenateDatasetOp::kDatasetType; in dataset_type()