Home
last modified time | relevance | path

Searched refs:interleave_depth (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Ddataset.h402 interleave_depth(ctx->interleave_depth()), in Params()
464 int64 interleave_depth = 0; member
532 int64 interleave_depth() { return params_.interleave_depth; } in interleave_depth() function
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_interleave_dataset_op.cc368 interleave_depth_ = ctx->interleave_depth(); in Initialize()
394 params.interleave_depth += 1; in Initialize()
446 (ctx->interleave_depth() == 0); in CreateNode()
1104 params.interleave_depth += 1; in InitializeInputs()
1402 params.interleave_depth += 1; in ReadElement()
Dprefetch_dataset_op.cc167 interleave_depth_ = ctx->interleave_depth(); in Initialize()
Dparallel_batch_dataset_op.cc205 interleave_depth_ = ctx->interleave_depth(); in Initialize()
Dparallel_filter_dataset_op.cc154 interleave_depth_ = ctx->interleave_depth(); in Initialize()
Dparallel_map_dataset_op.cc256 interleave_depth_ = ctx->interleave_depth(); in Initialize()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op.cc218 interleave_depth_ = ctx->interleave_depth(); in Initialize()