Home
last modified time | relevance | path

Searched refs:PaddedBatchDatasetOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dpadded_batch_dataset_op.cc36 /* static */ constexpr const char* const PaddedBatchDatasetOp::kDatasetType;
37 /* static */ constexpr const char* const PaddedBatchDatasetOp::kInputDataset;
38 /* static */ constexpr const char* const PaddedBatchDatasetOp::kBatchSize;
39 /* static */ constexpr const char* const PaddedBatchDatasetOp::kPaddedShapes;
40 /* static */ constexpr const char* const PaddedBatchDatasetOp::kPaddingValues;
41 /* static */ constexpr const char* const PaddedBatchDatasetOp::kDropRemainder;
42 /* static */ constexpr const char* const PaddedBatchDatasetOp::kParallelCopy;
43 /* static */ constexpr const char* const PaddedBatchDatasetOp::kToutputTypes;
44 /* static */ constexpr const char* const PaddedBatchDatasetOp::kOutputShapes;
45 /* static */ constexpr const char* const PaddedBatchDatasetOp::kNumPaddedShapes;
[all …]
Dpadded_batch_dataset_op_test.cc66 *input_names = {PaddedBatchDatasetOp::kInputDataset, in GetInputNames()
67 PaddedBatchDatasetOp::kBatchSize}; in GetInputNames()
71 strings::StrCat(PaddedBatchDatasetOp::kPaddedShapes, "_", i)); in GetInputNames()
76 strings::StrCat(PaddedBatchDatasetOp::kPaddingValues, "_", j)); in GetInputNames()
78 input_names->push_back(PaddedBatchDatasetOp::kDropRemainder); in GetInputNames()
84 {PaddedBatchDatasetOp::kParallelCopy, parallel_copy_}, in GetAttributes()
85 {PaddedBatchDatasetOp::kToutputTypes, output_dtypes_}, in GetAttributes()
86 {PaddedBatchDatasetOp::kOutputShapes, output_shapes_}, in GetAttributes()
87 {PaddedBatchDatasetOp::kNumPaddedShapes, num_padded_shapes_}}; in GetAttributes()
92 return PaddedBatchDatasetOp::kDatasetType; in dataset_type()
[all …]
Dpadded_batch_dataset_op.h23 class PaddedBatchDatasetOp : public UnaryDatasetOpKernel {
36 explicit PaddedBatchDatasetOp(OpKernelConstruction* ctx);