Searched refs:SetDimensionSizeInternal (Results 1 – 4 of 4) sorted by relevance
190 StatusOr<XlaOp> SetDimensionSizeInternal(const Shape& shape, XlaOp operand,
318 StatusOr<XlaOp> MlirHloBuilder::SetDimensionSizeInternal(const Shape& shape, in SetDimensionSizeInternal() function in xla::MlirHloBuilder
3371 return SetDimensionSizeInternal(shape, operand, val, dimension); in SetDimensionSize()3375 StatusOr<XlaOp> XlaBuilder::SetDimensionSizeInternal(const Shape& shape, in SetDimensionSizeInternal() function in xla::XlaBuilder
898 virtual StatusOr<XlaOp> SetDimensionSizeInternal(const Shape& shape,