Searched refs:dot_dimension_numbers_ (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction.h1107 CHECK(dot_dimension_numbers_ != nullptr);
1108 return *dot_dimension_numbers_;
1389 std::unique_ptr<DotDimensionNumbers> dot_dimension_numbers_; member in class:xla::HloInstruction
H A Dhlo_instruction.cc122 instruction->dot_dimension_numbers_ =
357 instruction->dot_dimension_numbers_ =
370 instruction->dot_dimension_numbers_ = MakeUnique<DotDimensionNumbers>();
371 instruction->dot_dimension_numbers_->add_lhs_contracting_dimensions(1);
372 instruction->dot_dimension_numbers_->add_rhs_contracting_dimensions(0);
1299 *dot_dimension_numbers_);
2213 if (dot_dimension_numbers_ != nullptr) {
2349 if (dot_dimension_numbers_ != nullptr) {
2350 *proto.mutable_dot_dimension_numbers() = *dot_dimension_numbers_;
3340 if (dot_dimension_numbers_
[all...]

Completed in 140 milliseconds