Searched refs:GetShapePtr (Results 1 – 5 of 5) sorted by relevance
204 TF_ASSIGN_OR_RETURN(const xla::Shape* shape, builder->GetShapePtr(x)); in operator >>()218 StatusOr<const Shape*> XlaBuilder::GetShapePtr(XlaOp op) const { in GetShapePtr() function in xla::XlaBuilder229 TF_ASSIGN_OR_RETURN(const Shape* shape, GetShapePtr(op)); in GetShape()237 TF_ASSIGN_OR_RETURN(const Shape* shape, GetShapePtr(operand)); in GetOperandShapes()568 TF_ASSIGN_OR_RETURN(const Shape* operand_shape, GetShapePtr(operand)); in AddBroadcastSequence()618 TF_ASSIGN_OR_RETURN(const Shape* operand_shape, GetShapePtr(operand)); in UnaryOp()630 TF_ASSIGN_OR_RETURN(const Shape* lhs_shape, GetShapePtr(lhs)); in BinaryOp()631 TF_ASSIGN_OR_RETURN(const Shape* rhs_shape, GetShapePtr(rhs)); in BinaryOp()669 GetShapePtr(updated_lhs)); in BinaryOp()675 GetShapePtr(updated_rhs)); in BinaryOp()[all …]
340 virtual StatusOr<const Shape*> GetShapePtr(XlaOp op) const;
102 StatusOr<const Shape*> GetShapePtr(XlaOp op) const override;
596 StatusOr<const Shape*> MlirHloBuilder::GetShapePtr(XlaOp op) const { in GetShapePtr() function in xla::MlirHloBuilder
469 builder->GetShapePtr(minval)); in ConvertRandomBitsToUniformFloatingPoint()470 TF_ASSIGN_OR_RETURN(const Shape* bits_shape, builder->GetShapePtr(bits)); in ConvertRandomBitsToUniformFloatingPoint()