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