Lines Matching refs:buffer_size
37 int64 buffer_size;
39 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size));
40 OP_REQUIRES(ctx, buffer_size > 0,
41 errors::InvalidArgument("buffer_size must be > 0"));
43 *output = new Dataset(ctx, input, buffer_size);
49 Dataset(OpKernelContext* ctx, const DatasetBase* input, int64 buffer_size)
50 : GraphDatasetBase(ctx), input_(input), buffer_size_(buffer_size) {
76 Node* buffer_size = nullptr;
77 TF_RETURN_IF_ERROR(b->AddScalar(buffer_size_, &buffer_size));
79 b->AddDataset(this, {input_graph_node, buffer_size}, output));
157 writer->WriteScalar(full_name("buffer_size"), buffer_.size()));
181 size_t buffer_size;
185 reader->ReadScalar(full_name("buffer_size"), &temp));
186 buffer_size = static_cast<size_t>(temp);
188 for (size_t i = 0; i < buffer_size; i++) {
