Searched defs:output_shape_index (Results 1 – 2 of 2) sorted by relevance
275 repeated int64 output_shape_index = 1; field
2645 ShapeIndex output_shape_index) { in EmitPrologueForOneReduction()2722 ShapeIndex output_shape_index; in EmitPrologueForReduction() local2897 ShapeIndex output_shape_index; in EmitTileElementForReduction() local