| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| cardinality_test.py | 83 lambda _: dataset_ops.Dataset.from_tensors(0), cycle_length=1),
87 cycle_length=1, variable in class:NumElementsTest
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| training_utils_test.py | 116 lambda _: dataset_ops.Dataset.from_tensors(0), cycle_length=1),
120 cycle_length=1, variable in class:DatasetUtilsTest
164 lambda _: dataset_ops.Dataset.from_tensors(0), cycle_length=1),
173 cycle_length=1,
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| interleave_dataset_op.cc | 43 OP_REQUIRES_OK(ctx, ctx->input("cycle_length", &cycle_length_t));
45 errors::InvalidArgument("cycle_length must be a scalar."));
46 const int64 cycle_length = cycle_length_t->flat<int64>()(0); variable
48 ctx, cycle_length > 0,
49 errors::InvalidArgument("cycle_length must be greater than zero."));
65 new Dataset(ctx, input, func_, std::move(captured_func), cycle_length,
74 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length,
81 cycle_length_(cycle_length),
|
| parallel_interleave_dataset_op.cc | 65 int64 cycle_length = 0; variable
67 ParseScalarArgument(ctx, "cycle_length", &cycle_length));
68 OP_REQUIRES(ctx, cycle_length > 0,
69 errors::InvalidArgument("`cycle_length` must be > 0"));
85 ctx, num_parallel_calls <= cycle_length,
87 "num_parallel_calls must less than or equal to cycle_length."));
100 cycle_length, block_length, num_parallel_calls, sloppy_,
109 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length,
117 cycle_length_(cycle_length),
[all...] |
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| parallel_interleave_dataset_op.cc | 48 int64 cycle_length = 0; variable
50 ParseScalarArgument(ctx, "cycle_length", &cycle_length));
51 OP_REQUIRES(ctx, cycle_length > 0,
52 errors::InvalidArgument("`cycle_length` must be > 0"));
84 cycle_length, block_length, sloppy, buffer_output_elements,
93 std::unique_ptr<CapturedFunction> captured_func, int64 cycle_length,
101 cycle_length_(cycle_length),
[all...] |
