Searched refs:dimension_number (Results 1 – 4 of 4) sorted by relevance
647 int64 dimension_number) { in GetDimension() argument648 return shape.dimensions(GetDimensionNumber(shape, dimension_number)); in GetDimension()652 int64 dimension_number) { in GetDimensionNumber() argument653 if (dimension_number < 0) { in GetDimensionNumber()654 dimension_number += Rank(shape); in GetDimensionNumber()656 CHECK_GE(dimension_number, 0); in GetDimensionNumber()657 return dimension_number; in GetDimensionNumber()
286 static int64 GetDimension(const Shape& shape, int64 dimension_number);296 static int64 GetDimensionNumber(const Shape& shape, int64 dimension_number);
61 for (int64 dimension_number : minor_to_major) { in MakeLayout() local62 layout.add_minor_to_major(dimension_number); in MakeLayout()
422 for (int64 dimension_number = 0; in InferConcatOpShape() local423 dimension_number < ShapeUtil::Rank(*arg_shape); ++dimension_number) { in InferConcatOpShape()424 if (arg_shape->dimensions(dimension_number) != in InferConcatOpShape()425 shape->dimensions(dimension_number)) { in InferConcatOpShape()426 if (dimension_number == dimension) { in InferConcatOpShape()