Home
last modified time | relevance | path

Searched refs:GetAsComplex128 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_test.cc2101 TEST_F(LiteralUtilTest, GetAsComplex128) { in TEST_F() argument
2104 EXPECT_EQ(*c1.GetAsComplex128({}), value); in TEST_F()
2106 EXPECT_EQ(*c2.GetAsComplex128({}), value); in TEST_F()
2109 EXPECT_EQ(*c4.GetAsComplex128({}), value); in TEST_F()
2112 EXPECT_EQ(*c5.GetAsComplex128({}), other_value); in TEST_F()
2114 EXPECT_FALSE(c6.GetAsComplex128({}).has_value()); in TEST_F()
Dliteral.h158 complex128 as_complex128 = *GetAsComplex128(multi_index); in IsEqualAt()
167 auto as_complex128 = GetAsComplex128(multi_index); in IsEqualAt()
185 std::optional<complex128> GetAsComplex128(
Dliteral.cc1216 std::optional<complex128> LiteralBase::GetAsComplex128( in GetAsComplex128() function in xla::LiteralBase
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_rewriter.cc179 *alpha->literal().GetAsComplex128({}) * prev_alpha; in HandleMultiply()