Searched defs:dimension_number (Results 1 – 3 of 3) sorted by relevance
63 for (int64 dimension_number : minor_to_major) { in MakeLayout() local
565 int64 dimension_number) { in GetDimension()570 int64 dimension_number) { in GetDimensionNumber()
387 for (int64 dimension_number = 0; dimension_number < arg_shape->rank(); in InferConcatOpShape() local