HomeSort by relevance Sort by last modified time
    Searched refs:backward_conv_dnums (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc 189 ConvolutionDimensionNumbers backward_conv_dnums; local
190 backward_conv_dnums.set_input_batch_dimension(input_feature_dim);
191 backward_conv_dnums.set_input_feature_dimension(input_batch_dim);
193 backward_conv_dnums.add_input_spatial_dimensions(input_spatial_dims[i]);
195 backward_conv_dnums.set_output_batch_dimension(kernel_input_feature_dim);
196 backward_conv_dnums.set_output_feature_dimension(kernel_output_feature_dim);
198 backward_conv_dnums.add_output_spatial_dimensions(kernel_spatial_dims[i]);
205 backward_conv_dnums.set_kernel_input_feature_dimension(output_batch_dim);
206 backward_conv_dnums.set_kernel_output_feature_dimension(output_feature_dim);
208 backward_conv_dnums.add_kernel_spatial_dimensions(output_spatial_dims[i])
    [all...]
pad_insertion.cc 221 ConvolutionDimensionNumbers backward_conv_dnums = local
234 int64 dim = backward_conv_dnums.input_spatial_dimensions(i);
266 backward_conv_dnums);
284 ConvolutionDimensionNumbers backward_conv_dnums = local
318 int64 dim = backward_conv_dnums.output_spatial_dimensions(i);
331 backward_conv_dnums);
355 int64 dim = backward_conv_dnums.output_spatial_dimensions(i);

Completed in 216 milliseconds