Searched refs:output_dim_is_batch_dims_ (Results 1 – 1 of 1) sorted by relevance
1504 output_dim_is_batch_dims_.push_back( in OutputBatchIndexToInputIndex()1559 if (!output_dim_is_batch_dims_[i]) { in PropagateOutputIndexGatherDimsToIndexVectorIndex()1603 std::vector<bool> output_dim_is_batch_dims_; member in xla::OutputBatchIndexToInputIndex