Lines Matching defs:block_size_
52 OP_REQUIRES_OK(context, context->GetAttr("block_size", &block_size_));
53 OP_REQUIRES(context, block_size_ > 1,
55 block_size_));
92 const int block_size_sq = block_size_ * block_size_;
94 // The depth must be divisible by block_size_ * block_size_
101 const int output_width = input_width * block_size_;
102 const int output_height = input_height * block_size_;
121 functor(context->eigen_device<GPUDevice>(), Tinput_v, block_size_,
126 functor(context->eigen_device<GPUDevice>(), Tinput, block_size_,
137 functor(context->eigen_device<Device>(), Tinput, block_size_, Toutput);
142 int block_size_;
