Searched refs:reduction_output_shape_indices_ (Results 1 – 1 of 1) sorted by relevance
2569 return &reduction_output_shape_indices_; in GetMutableReductionOutputShapeIndices()2572 return reduction_output_shape_indices_; in GetReductionOutputShapeIndices()2610 InlinedVector<ShapeIndex, 1> reduction_output_shape_indices_; member in xla::gpu::ReductionCodegenInfo