Searched refs:filter_param (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 443 const Tensor& filter_param, int32 row_stride, int32 col_stride, in launch() argument
478 const int output_depth = GetFilterDim(filter_param, filter_format, 'O'); in launch()
479 const int filter_rows = GetFilterDim(filter_param, filter_format, 'H'); in launch()
480 const int filter_cols = GetFilterDim(filter_param, filter_format, 'W'); in launch()
620 const Tensor* filter = &filter_param; in launch()
629 FORMAT_OIHW, filter_param.shape(), FORMAT_HWIO), in launch()
633 To32Bit(filter_param.tensor<T, 4>()), in launch()
|