tensorflow/compiler/mlir/xla/hlo_function_importer.cc
// Returns whether the instruction is a default dot operation.
Returns whether the instruction is a default dot operation.
Returns whether the instruction is a default dot operation .
bool DotIsDefault(const HloInstruction* instruction) {
  auto dnums = instruction->dot_dimension_numbers();
  DotDimensionNumbers default_dimension_numbers;
  default_dimension_numbers.add_lhs_contracting_dimensions(
      instruction->operand(0)->shape().dimensions_size() == 1 ? 0 : 1);
  default_dimension_numbers.add_rhs_contracting_dimensions(0);
  return xla::protobuf_util::ProtobufEquals(dnums, default_dimension_numbers);
}
bool DotIsDefault ( const HloInstruction * instruction ) { auto dnums = instruction -> dot_dimension_numbers ( ) ; DotDimensionNumbers default_dimension_numbers ; default_dimension_numbers . add_lhs_contracting_dimensions ( instruction -> operand ( 0 ) -> shape ( ) . dimensions_size ( ) == 1 ? 0 : 1 ) ; default_dimension_numbers . add_rhs_contracting_dimensions ( 0 ) ; return xla :: protobuf_util :: ProtobufEquals ( dnums , default_dimension_numbers ) ; }
Returns whether the instruction is a default dot operation.
Returns whether the instruction is a default dot operation .