Searched refs:conv_input_param (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 442 const Tensor& conv_input_param, ScaleType conv_input_scale, in launch() argument
470 const int batch_size = GetTensorDim(conv_input_param, data_format, 'N'); in launch()
471 int conv_input_rows = GetTensorDim(conv_input_param, data_format, 'H'); in launch()
472 int conv_input_cols = GetTensorDim(conv_input_param, data_format, 'W'); in launch()
475 GetTensorDim(conv_input_param, data_format, 'C') * vect; in launch()
483 const Tensor* conv_input = &conv_input_param; in launch()
526 To32Bit(conv_input_param.reinterpret_last_dimension<VectT, 4>()); in launch()
|