Home | History | Annotate | Download | only in data

Lines Matching refs:buffer_size

47     int64 buffer_size = -1;
49 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size));
51 ctx, buffer_size >= 0,
52 errors::InvalidArgument("`buffer_size` must be >= 0 (0 == default)"));
65 if (buffer_size != 0) {
67 zlib_compression_options.input_buffer_size = buffer_size;
116 Node* buffer_size = nullptr;
120 b->AddScalar(options_.input_buffer_size, &buffer_size));
122 this, {filenames, compression_type, buffer_size}, output));
300 int64 buffer_size = -1;
302 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size));
303 OP_REQUIRES(ctx, buffer_size >= 0,
304 errors::InvalidArgument("`buffer_size` must be >= 0"));
305 if (buffer_size == 0) {
306 buffer_size = 256 << 10; // 256 kB as default.
310 footer_bytes, buffer_size);
318 int64 buffer_size)
324 buffer_size_(buffer_size) {}
354 Node* buffer_size = nullptr;
359 TF_RETURN_IF_ERROR(b->AddScalar(buffer_size_, &buffer_size));
362 {filenames, header_bytes, record_bytes, footer_bytes, buffer_size},
520 int64 buffer_size = -1;
522 ctx, ParseScalarArgument<int64>(ctx, "buffer_size", &buffer_size));
523 OP_REQUIRES(ctx, buffer_size >= 0,
525 "`buffer_size` must be >= 0 (0 == no buffering)"));
528 new Dataset(ctx, std::move(filenames), compression_type, buffer_size);
535 const string& compression_type, int64 buffer_size)
541 if (buffer_size > 0) {
542 options_.buffer_size = buffer_size;
572 Node* buffer_size = nullptr;
573 TF_RETURN_IF_ERROR(b->AddScalar(options_.buffer_size, &buffer_size));
575 this, {filenames, compression_type, buffer_size}, output));