Searched refs:gather_dimension_numbers_ (Results 1 – 2 of 2) sorted by relevance
1660 CHECK(gather_dimension_numbers_ != nullptr); in gather_dimension_numbers()1661 return *gather_dimension_numbers_; in gather_dimension_numbers()1693 std::unique_ptr<GatherDimensionNumbers> gather_dimension_numbers_; variable
2731 gather_dimension_numbers_ = in HloGatherInstruction()