Searched refs:gather_dimension_numbers_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.cc | 2316 gather_dimension_numbers_ = in HloGatherInstruction()
2322 CHECK(gather_dimension_numbers_ != nullptr); in GatherDimensionNumbersToString()
2325 StrJoin(gather_dimension_numbers_->offset_dims(), ","), "}"); in GatherDimensionNumbersToString()
2328 StrJoin(gather_dimension_numbers_->collapsed_slice_dims(), ","), "}"); in GatherDimensionNumbersToString()
2331 StrJoin(gather_dimension_numbers_->start_index_map(), ","), "}"); in GatherDimensionNumbersToString()
2333 "index_vector_dim=", gather_dimension_numbers_->index_vector_dim()); in GatherDimensionNumbersToString()
|
| D | hlo_instructions.h | 1367 CHECK(gather_dimension_numbers_ != nullptr); in gather_dimension_numbers()
1368 return *gather_dimension_numbers_; in gather_dimension_numbers()
1395 std::unique_ptr<GatherDimensionNumbers> gather_dimension_numbers_; variable
|