Lines Matching refs:buffer_size
45 int64 buffer_size, int64 count)
48 buffer_size_(buffer_size),
316 int64 buffer_size;
318 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size));
320 ctx, buffer_size > 0,
321 errors::InvalidArgument("buffer_size must be greater than zero."));
339 new ReshufflingDataset(ctx, input, buffer_size, seed, seed2, count);
342 new FixedSeedDataset(ctx, input, buffer_size, seed, seed2, count);
352 int64 buffer_size, int64 seed, int64 seed2, int64 count)
353 : ShuffleDatasetBase(ctx, input, buffer_size, count),
392 int64 buffer_size, int64 seed, int64 seed2, int64 count)
393 : ShuffleDatasetBase(ctx, input, buffer_size, count),
413 Node* buffer_size = nullptr;
418 TF_RETURN_IF_ERROR(b->AddScalar(buffer_size_, &buffer_size));
423 this, {input_graph_node, buffer_size
445 int64 buffer_size;
447 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size));
449 ctx, buffer_size > 0,
450 errors::InvalidArgument("buffer_size must be greater than zero."));
468 *output = new Dataset(ctx, input, buffer_size, seed, seed2, count);
474 Dataset(OpKernelContext* ctx, const DatasetBase* input, int64 buffer_size,
476 : ShuffleDatasetBase(ctx, input, buffer_size, count),
497 Node* buffer_size = nullptr;
502 TF_RETURN_IF_ERROR(b->AddScalar(buffer_size_, &buffer_size));
507 this, {input_graph_node, buffer_size, seed, seed2, count}, // Inputs
