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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
conv_canonicalization.cc 39 auto input_batch_dim = dnums.input_batch_dimension(); local
60 new_input_dim_order[0] = input_batch_dim;
61 new_input_dims[0] = input->shape().dimensions(input_batch_dim);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc 81 auto input_batch_dim = conv_dnums.input_batch_dimension(); local
114 if (input_batch_dim == output_batch_dim &&
191 backward_conv_dnums.set_input_feature_dimension(input_batch_dim);

Completed in 106 milliseconds