Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_test.cc561 TEST_F(LiteralUtilTest, IsAllFloat) { in TEST_F() argument
563 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(false).IsAllFloat(0)); in TEST_F()
564 EXPECT_FALSE(LiteralUtil::CreateR0<int8>(0).IsAllFloat(0)); in TEST_F()
565 EXPECT_FALSE(LiteralUtil::CreateR0<uint8>(0).IsAllFloat(0)); in TEST_F()
566 EXPECT_FALSE(LiteralUtil::CreateR0<int>(0).IsAllFloat(0)); in TEST_F()
568 EXPECT_TRUE(LiteralUtil::CreateR0<float>(0).IsAllFloat(0)); in TEST_F()
569 EXPECT_TRUE(LiteralUtil::CreateR0<float>(.5).IsAllFloat(.5)); in TEST_F()
570 EXPECT_TRUE(LiteralUtil::CreateR0<float>(-.5).IsAllFloat(-.5)); in TEST_F()
571 EXPECT_FALSE(LiteralUtil::CreateR0<float>(-.5).IsAllFloat(-.49)); in TEST_F()
573 LiteralUtil::CreateR2<float>({{0, 0, 0}, {0, .1, 0}}).IsAllFloat(0)); in TEST_F()
[all …]
Dliteral.h189 bool IsAllFloat(float value) const;
Dliteral.cc1623 bool LiteralBase::IsAllFloat(float value) const { in IsAllFloat() function in xla::LiteralBase
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc62 operand->literal().IsAllFloat(value)) { in IsFPLiteralWithValue()