Searched refs:conv_input_cols (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 209 const int32 conv_input_cols = GetTensorDim(conv_input, data_format_, 'W'); in Compute() local
215 OP_REQUIRES_OK(context, GetWindowedOutputSize(conv_input_cols, filter_cols, in Compute()
225 << conv_input_cols << ", conv_input_rows = " << conv_input_rows in Compute()
472 int conv_input_cols = GetTensorDim(conv_input_param, data_format, 'W'); in launch() local
498 0, (output_cols - 1) * col_stride + filter_cols - conv_input_cols); in launch()
513 conv_input_cols + extra_left_padding + extra_right_padding; in launch()
538 conv_input_cols = new_conv_input_cols; in launch()
553 conv_input_cols, conv_input_depth, in launch()
585 .set_width(conv_input_cols) in launch()
666 {{conv_input_rows, conv_input_cols}}, in launch()
|