Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_rewriter.cc89 *alpha->literal().GetAsComplex128({}) * prev_alpha; in HandleMultiply()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h153 complex128 as_complex128 = *GetAsComplex128(multi_index); in IsEqualAt()
161 auto as_complex128 = GetAsComplex128(multi_index); in IsEqualAt()
178 absl::optional<complex128> GetAsComplex128(
Dliteral_test.cc2092 TEST_F(LiteralUtilTest, GetAsComplex128) { in TEST_F() argument
2095 EXPECT_EQ(*c1.GetAsComplex128({}), value); in TEST_F()
2097 EXPECT_EQ(*c2.GetAsComplex128({}), value); in TEST_F()
2100 EXPECT_EQ(*c4.GetAsComplex128({}), value); in TEST_F()
2103 EXPECT_EQ(*c5.GetAsComplex128({}), other_value); in TEST_F()
2105 EXPECT_FALSE(c6.GetAsComplex128({}).has_value()); in TEST_F()
Dliteral.cc1062 absl::optional<complex128> LiteralBase::GetAsComplex128( in GetAsComplex128() function in xla::LiteralBase