Searched refs:ParallelInterleaveDatasetOp (Results 1 – 6 of 6) sorted by relevance
70 input_names->emplace_back(ParallelInterleaveDatasetOp::kInputDataset); in GetInputNames()73 absl::StrCat(ParallelInterleaveDatasetOp::kOtherArguments, "_", i)); in GetInputNames()75 input_names->emplace_back(ParallelInterleaveDatasetOp::kCycleLength); in GetInputNames()76 input_names->emplace_back(ParallelInterleaveDatasetOp::kBlockLength); in GetInputNames()78 ParallelInterleaveDatasetOp::kBufferOutputElements); in GetInputNames()80 ParallelInterleaveDatasetOp::kPrefetchInputElements); in GetInputNames()95 return ParallelInterleaveDatasetOp::kDatasetType; in dataset_type()421 name_utils::OpName(ParallelInterleaveDatasetOp::kDatasetType, params))); in TEST_F()472 name_utils::IteratorPrefix(ParallelInterleaveDatasetOp::kDatasetType, in TEST_F()
44 ParallelInterleaveDatasetOp::kDatasetType;46 ParallelInterleaveDatasetOp::kInputDataset;48 ParallelInterleaveDatasetOp::kOtherArguments;50 ParallelInterleaveDatasetOp::kCycleLength;52 ParallelInterleaveDatasetOp::kBlockLength;54 ParallelInterleaveDatasetOp::kDeterministic;55 /* static */ constexpr const char* const ParallelInterleaveDatasetOp::kSloppy;57 ParallelInterleaveDatasetOp::kBufferOutputElements;59 ParallelInterleaveDatasetOp::kPrefetchInputElements;60 /* static */ constexpr const char* const ParallelInterleaveDatasetOp::kFunc;[all …]
29 class ParallelInterleaveDatasetOp : public UnaryDatasetOpKernel {47 explicit ParallelInterleaveDatasetOp(OpKernelConstruction* ctx);
60 ParallelInterleaveDatasetOp::kDatasetType;62 ParallelInterleaveDatasetOp::kInputDataset;64 ParallelInterleaveDatasetOp::kOtherArguments;66 ParallelInterleaveDatasetOp::kCycleLength;68 ParallelInterleaveDatasetOp::kBlockLength;70 ParallelInterleaveDatasetOp::kBufferOutputElements;72 ParallelInterleaveDatasetOp::kPrefetchInputElements;74 ParallelInterleaveDatasetOp::kNumParallelCalls;75 /* static */ constexpr const char* const ParallelInterleaveDatasetOp::kFunc;77 ParallelInterleaveDatasetOp::kTarguments;[all …]
77 input_names->emplace_back(ParallelInterleaveDatasetOp::kInputDataset); in GetInputNames()80 absl::StrCat(ParallelInterleaveDatasetOp::kOtherArguments, "_", i)); in GetInputNames()82 input_names->emplace_back(ParallelInterleaveDatasetOp::kCycleLength); in GetInputNames()83 input_names->emplace_back(ParallelInterleaveDatasetOp::kBlockLength); in GetInputNames()85 ParallelInterleaveDatasetOp::kBufferOutputElements); in GetInputNames()87 ParallelInterleaveDatasetOp::kPrefetchInputElements); in GetInputNames()88 input_names->emplace_back(ParallelInterleaveDatasetOp::kNumParallelCalls); in GetInputNames()103 return ParallelInterleaveDatasetOp::kDatasetType; in dataset_type()735 name_utils::OpName(ParallelInterleaveDatasetOp::kDatasetType, params))); in TEST_F()796 name_utils::IteratorPrefix(ParallelInterleaveDatasetOp::kDatasetType, in TEST_F()
25 class ParallelInterleaveDatasetOp : public UnaryDatasetOpKernel {44 explicit ParallelInterleaveDatasetOp(OpKernelConstruction* ctx);