Searched refs:dim_numbers_ (Results 1 – 2 of 2) sorted by relevance
1502 : dim_numbers_(*dim_numbers), start_indices_(*start_indices) { in OutputBatchIndexToInputIndex()1505 !absl::c_binary_search(dim_numbers_.offset_dims(), i)); in OutputBatchIndexToInputIndex()1510 std::distance(dim_numbers_.start_index_map().begin(), in OutputBatchIndexToInputIndex()1511 absl::c_find(dim_numbers_.start_index_map(), i)); in OutputBatchIndexToInputIndex()1513 dim_numbers_.start_index_map_size()) { in OutputBatchIndexToInputIndex()1524 start_indices_.shape().dimensions(dim_numbers_.index_vector_dim()); in OutputBatchIndexToInputIndex()1563 if (index_vector_index_i == dim_numbers_.index_vector_dim()) { in PropagateOutputIndexGatherDimsToIndexVectorIndex()1574 int64 index_vector_dim = dim_numbers_.index_vector_dim(); in FetchIndexVector()1616 const GatherDimensionNumbers& dim_numbers_; member in xla::OutputBatchIndexToInputIndex
2131 : dim_numbers_(*dim_numbers), scatter_indices_(*scatter_indices) {2134 !absl::c_binary_search(dim_numbers_.update_window_dims(), i));2139 FindIndex(dim_numbers_.scatter_dims_to_operand_dims(), i);2141 dim_numbers_.scatter_dims_to_operand_dims_size()) {2152 scatter_indices_.shape().dimensions(dim_numbers_.index_vector_dim());2191 if (index_vector_index_i == dim_numbers_.index_vector_dim()) {2202 int64 index_vector_dim = dim_numbers_.index_vector_dim();2243 const ScatterDimensionNumbers& dim_numbers_;