Home
last modified time | relevance | path

Searched refs:GetShapePtr (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc204 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::XlaBuilder
229 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 …]
Dxla_builder.h340 virtual StatusOr<const Shape*> GetShapePtr(XlaOp op) const;
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.h102 StatusOr<const Shape*> GetShapePtr(XlaOp op) const override;
Dmlir_hlo_builder.cc596 StatusOr<const Shape*> MlirHloBuilder::GetShapePtr(XlaOp op) const { in GetShapePtr() function in xla::MlirHloBuilder
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.cc469 builder->GetShapePtr(minval)); in ConvertRandomBitsToUniformFloatingPoint()
470 TF_ASSIGN_OR_RETURN(const Shape* bits_shape, builder->GetShapePtr(bits)); in ConvertRandomBitsToUniformFloatingPoint()