Lines Matching defs:block_size_
37 OP_REQUIRES_OK(ctx, ctx->GetAttr("block_size", &block_size_));
39 ctx, block_size_ > 1,
40 errors::InvalidArgument("Block size should be > 1: ", block_size_));
72 reshaped_shape.push_back(block_size_);
73 block_elems *= block_size_;
86 output_shape.push_back(input_shape[1 + i] * block_size_);
94 reshaped_shape.push_back(block_size_);
95 block_elems *= block_size_;
112 output_shape.push_back(input_shape[2 + i] * block_size_);
123 // block_size_,
124 // block_size_,
125 // depth / (block_size_ * block_size_)]
127 input_shape[feature_dim] % (block_size_ * block_size_) == 0,
131 block_size_, ")"));
140 // block_size_,
142 // block_size_,
143 // depth / (block_size_ * block_size_)]
151 // input_shape[1] * block_size_,
152 // input_shape[2] * block_size_,
153 // depth / (block_size_ * block_size_)]
163 int block_size_;
