Searched defs:dimension_number (Results 1 – 4 of 4) sorted by relevance
63 for (int64 dimension_number : minor_to_major) { in MakeLayout() local
684 int64 dimension_number) { in GetDimension()689 int64 dimension_number) { in GetDimensionNumber()
379 const DotDimensionNumbers& dimension_number, in DotGeneralInternal()
413 for (int64 dimension_number = 0; dimension_number < arg_shape->rank(); in InferConcatOpShape() local