Home
last modified time | relevance | path

Searched refs:kPrefetchInputElements (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dparallel_interleave_dataset_op.h34 static constexpr const char* const kPrefetchInputElements = variable
H A Dparallel_interleave_dataset_op_test.cc87 ParallelInterleaveDatasetOp::kPrefetchInputElements); in GetInputNames()
H A Dparallel_interleave_dataset_op.cc72 ParallelInterleaveDatasetOp::kPrefetchInputElements; member in tensorflow::data::ParallelInterleaveDatasetOp
1725 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kPrefetchInputElements, in MakeDataset()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dparallel_interleave_dataset_op.h40 static constexpr const char* const kPrefetchInputElements = variable
H A Dparallel_interleave_dataset_op_test.cc80 ParallelInterleaveDatasetOp::kPrefetchInputElements); in GetInputNames()
H A Dparallel_interleave_dataset_op.cc59 ParallelInterleaveDatasetOp::kPrefetchInputElements; member in tensorflow::data::experimental::ParallelInterleaveDatasetOp
1265 OP_REQUIRES_OK(ctx, ParseScalarArgument(ctx, kPrefetchInputElements, in MakeDataset()