Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util_test.cc461 TEST_F(LiteralUtilTest, IsAllFloat) { in TEST_F() argument
463 EXPECT_FALSE(Literal::CreateR0<bool>(false)->IsAllFloat(0)); in TEST_F()
464 EXPECT_FALSE(Literal::CreateR0<int8>(0)->IsAllFloat(0)); in TEST_F()
465 EXPECT_FALSE(Literal::CreateR0<uint8>(0)->IsAllFloat(0)); in TEST_F()
466 EXPECT_FALSE(Literal::CreateR0<int>(0)->IsAllFloat(0)); in TEST_F()
468 EXPECT_TRUE(Literal::CreateR0<float>(0)->IsAllFloat(0)); in TEST_F()
469 EXPECT_TRUE(Literal::CreateR0<float>(.5)->IsAllFloat(.5)); in TEST_F()
470 EXPECT_TRUE(Literal::CreateR0<float>(-.5)->IsAllFloat(-.5)); in TEST_F()
471 EXPECT_FALSE(Literal::CreateR0<float>(-.5)->IsAllFloat(-.49)); in TEST_F()
473 Literal::CreateR2<float>({{0, 0, 0}, {0, .1, 0}})->IsAllFloat(0)); in TEST_F()
[all …]
Dliteral_util.h590 bool IsAllFloat(float value) const;
Dliteral_util.cc1534 bool Literal::IsAllFloat(float value) const { in IsAllFloat() function in xla::Literal
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc59 operand->literal().IsAllFloat(value); in IsFPLiteralWithValue()