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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
conv_canonicalization.cc 40 auto input_feature_dim = dnums.input_feature_dimension(); local
67 new_input_dim_order[num_dims - 1] = input_feature_dim;
69 input->shape().dimensions(input_feature_dim);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc 82 auto input_feature_dim = conv_dnums.input_feature_dimension(); local
190 backward_conv_dnums.set_input_batch_dimension(input_feature_dim);

Completed in 127 milliseconds