Home
last modified time | relevance | path

Searched refs:output_param (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc447 FilterTensorFormat filter_format, Tensor* output_param) { in launch() argument
476 const int output_rows = GetTensorDim(*output_param, data_format, 'H'); in launch()
477 const int output_cols = GetTensorDim(*output_param, data_format, 'W'); in launch()
545 Tensor* output = output_param; in launch()
776 output_param->tensor<T, 4>()); in launch()