Searched refs:dimension_number (Results 1 – 6 of 6) sorted by relevance
512 int64 dimension_number) { in GetDimension() argument513 return shape.dimensions(GetDimensionNumber(shape, dimension_number)); in GetDimension()517 int64 dimension_number) { in GetDimensionNumber() argument518 if (dimension_number < 0) { in GetDimensionNumber()519 dimension_number += shape.rank(); in GetDimensionNumber()521 CHECK_GE(dimension_number, 0); in GetDimensionNumber()522 return dimension_number; in GetDimensionNumber()
61 for (int64 dimension_number : minor_to_major) { in MakeLayout() local62 layout.add_minor_to_major(dimension_number); in MakeLayout()
324 static int64 GetDimension(const Shape& shape, int64 dimension_number);334 static int64 GetDimensionNumber(const Shape& shape, int64 dimension_number);
379 for (int64 dimension_number = 0; dimension_number < arg_shape->rank(); in InferConcatOpShape() local380 ++dimension_number) { in InferConcatOpShape()381 if (arg_shape->dimensions(dimension_number) != in InferConcatOpShape()382 shape->dimensions(dimension_number)) { in InferConcatOpShape()383 if (dimension_number == dimension) { in InferConcatOpShape()
215 int64 dimension_number = instr.dimensions(0); in IsConstantVisitor() local219 if (operand_shape.is_dynamic_dimension(dimension_number)) { in IsConstantVisitor()
779 const DotDimensionNumbers& dimension_number,