Searched refs:BitcastConvert (Results 1 – 9 of 9) sorted by relevance
33 m::BitcastConvert(param).WithShape(m::Shape().WithElementType(S32)); in IsNanSafeGt()35 m::BitcastConvert(param).WithShape(m::Shape().WithElementType(U32)); in IsNanSafeGt()38 m::BitcastConvert( in IsNanSafeGt()49 m::BitcastConvert(param).WithShape(m::Shape().WithElementType(S32)); in IsNanSafeGt()51 m::BitcastConvert(param).WithShape(m::Shape().WithElementType(U32)); in IsNanSafeGt()54 m::BitcastConvert( in IsNanSafeGt()
1808 "BitcastConvert", in CreateTestCases()1809 R"(HloModule BitcastConvert in CreateTestCases()
2007 XLA_UNOP_PATTERN(BitcastConvert)
294 parent_->GetEvaluatedLiteralFor(operand).BitcastConvert(
1115 m::BitcastConvert(param).WithShape(m::Shape().WithElementType(S32)); in IsNanSafeGt()1117 m::BitcastConvert(param).WithShape(m::Shape().WithElementType(U32)); in IsNanSafeGt()1120 m::BitcastConvert( in IsNanSafeGt()1131 m::BitcastConvert(param).WithShape(m::Shape().WithElementType(S32)); in IsNanSafeGt()1133 m::BitcastConvert(param).WithShape(m::Shape().WithElementType(U32)); in IsNanSafeGt()1136 m::BitcastConvert( in IsNanSafeGt()
1499 TEST_F(LiteralUtilTest, BitcastConvert) { in TEST_F() argument1505 TF_ASSERT_OK_AND_ASSIGN(Literal converted, original.BitcastConvert(F32)); in TEST_F()1510 Status status = literal.BitcastConvert(F64).status(); in TEST_F()
264 StatusOr<Literal> BitcastConvert(PrimitiveType primitive_dest_type) const;
1520 StatusOr<Literal> LiteralBase::BitcastConvert( in BitcastConvert() function in xla::LiteralBase
864 string summary = "BitcastConvert operator";