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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
conv_canonicalization.cc 40 auto input_batch_dim = dnums.input_batch_dimension(); local
61 new_input_dim_order[0] = input_batch_dim;
62 new_input_dims[0] = input->shape().dimensions(input_batch_dim);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_conv_rewriter.cc 118 auto input_batch_dim = conv_dnums.input_batch_dimension(); local
151 if (input_batch_dim == output_batch_dim &&
231 backward_conv_dnums.set_input_feature_dimension(input_batch_dim);
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_cost_analysis.cc 444 const int64 input_batch_dim = dnums.input_batch_dimension(); local
451 const int64 batch = ShapeUtil::GetDimension(lhs_shape, input_batch_dim);
hlo_evaluator_typed_visitor.h     [all...]
  /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
convert_nodes.cc     [all...]

Completed in 219 milliseconds