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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
conv_canonicalization.cc 102 auto output_feature_dim = dnums.output_feature_dimension(); local
110 new_output_dim_order[num_dims - 1] = output_feature_dim;
111 new_conv_dims[num_dims - 1] = hlo->shape().dimensions(output_feature_dim);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc 88 auto output_feature_dim = conv_dnums.output_feature_dimension(); local
206 backward_conv_dnums.set_kernel_output_feature_dimension(output_feature_dim);

Completed in 263 milliseconds