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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
conv_canonicalization.cc 101 auto output_batch_dim = dnums.output_batch_dimension(); local
103 new_output_dim_order[0] = output_batch_dim;
104 new_conv_dims[0] = hlo->shape().dimensions(output_batch_dim);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc 87 auto output_batch_dim = conv_dnums.output_batch_dimension(); local
114 if (input_batch_dim == output_batch_dim &&
205 backward_conv_dnums.set_kernel_input_feature_dimension(output_batch_dim);
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator.cc     [all...]

Completed in 96 milliseconds