Searched refs:ReportErrorOrReturn (Results 1 – 17 of 17) sorted by relevance
37 return b.ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in DoWithUpcastToF32()69 return b.ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in IsPosInf()80 return b.ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in IsNegInf()91 return b.ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in IsInf()99 return b.ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in IsNan()107 return b.ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in IsNegZero()207 return b.ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in Erfc()225 return b.ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in Erf()283 return b.ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in ErfInv()409 return b.ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in Lgamma()[all …]
24 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in SliceInMinorDims()54 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in UpdateSlice()72 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in UpdateSliceInMinorDims()112 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in DynamicSliceInMinorDims()131 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in DynamicUpdateSliceInMinorDims()139 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in TorchGather()166 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in TorchIndexSelect()
51 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in GetMatrixDiagonal()93 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in TriangleMask()164 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in Einsum()257 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in BatchDot()331 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in Einsum()341 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in TransposeInMinorDims()
78 return b->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in ComputeSums()137 return b->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in MaxPool()153 return b->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in AvgPool()199 return b->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in AvgPoolGrad()
87 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in ScalarLike()101 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in FullLike()
34 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in ZerosLike()122 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in NanValue()
115 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in Any()130 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in ArgMinMax()
26 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in TopK()
86 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> {
89 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in operator >>()139 XlaOp XlaBuilder::ReportErrorOrReturn(const StatusOr<XlaOp>& op) { in ReportErrorOrReturn() function in xla::XlaBuilder149 XlaOp XlaBuilder::ReportErrorOrReturn( in ReportErrorOrReturn() function in xla::XlaBuilder151 return ReportErrorOrReturn(op_creator()); in ReportErrorOrReturn()472 return ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in UnaryOp()485 return ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in BinaryOp()555 return ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in TernaryOp()590 return ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in ConstantLiteral()599 return ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in Iota()613 return ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in Call()[all …]
408 Add(b.ReportErrorOrReturn(op), ConstantR0<float>(&b, 2.0)); in TEST_F()417 Add(b.ReportErrorOrReturn(op), ConstantR0<float>(&b, 2.0)); in TEST_F()
269 XlaOp ReportErrorOrReturn(const StatusOr<XlaOp>& op);273 XlaOp ReportErrorOrReturn(const std::function<StatusOr<XlaOp>()>& op_creator);
41 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in DiagonalBlocks()128 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in InvertDiagonalBlocks()257 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in SolveWithInvertedDiagonalBlocks()355 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in BuildTriangularSolve()
56 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in CholeskyUnblocked()141 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in BuildCholesky()
79 context->SetOutput(0, builder->ReportErrorOrReturn(result)); in Compile()
81 return builder->ReportErrorOrReturn([&]() -> xla::StatusOr<xla::XlaOp> { in AsXlaOp()
723 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in QR()731 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in Eigh()739 return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { in SVD()