Searched refs:conv_input_rows (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 208 const int32 conv_input_rows = GetTensorDim(conv_input, data_format_, 'H'); in Compute() local
212 OP_REQUIRES_OK(context, GetWindowedOutputSize(conv_input_rows, filter_rows, in Compute()
225 << conv_input_cols << ", conv_input_rows = " << conv_input_rows in Compute()
471 int conv_input_rows = GetTensorDim(conv_input_param, data_format, 'H'); in launch() local
496 0, (output_rows - 1) * row_stride + filter_rows - conv_input_rows); in launch()
511 conv_input_rows + extra_top_padding + extra_bottom_padding; in launch()
537 conv_input_rows = new_conv_input_rows; in launch()
552 ctx, *conv_input, batch_size, conv_input_rows, in launch()
584 .set_height(conv_input_rows) in launch()
666 {{conv_input_rows, conv_input_cols}}, in launch()
|