Searched refs:IsAllFirst (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 600 TEST_F(LiteralUtilTest, IsAllFirst) { in TEST_F() argument
602 EXPECT_FALSE(LiteralUtil::CreateR1<bool>({false, true}).IsAllFirst()); in TEST_F()
603 EXPECT_TRUE(LiteralUtil::CreateR1<bool>({false, false}).IsAllFirst()); in TEST_F()
604 EXPECT_FALSE(LiteralUtil::CreateR1<int8>({1, 1, 2}).IsAllFirst()); in TEST_F()
605 EXPECT_TRUE(LiteralUtil::CreateR1<int8>({5, 5, 5, 5}).IsAllFirst()); in TEST_F()
606 EXPECT_FALSE(LiteralUtil::CreateR1<uint8>({1, 1, 2}).IsAllFirst()); in TEST_F()
607 EXPECT_TRUE(LiteralUtil::CreateR1<int32>({5, 5, 5, 5}).IsAllFirst()); in TEST_F()
608 EXPECT_FALSE(LiteralUtil::CreateR1<int32>({1, 1, 2}).IsAllFirst()); in TEST_F()
609 EXPECT_TRUE(LiteralUtil::CreateR1<uint32>({5, 5, 5, 5}).IsAllFirst()); in TEST_F()
610 EXPECT_FALSE(LiteralUtil::CreateR1<uint32>({1, 1, 2}).IsAllFirst()); in TEST_F()
[all …]
|
| D | literal.h | 234 bool IsAllFirst() const;
|
| D | literal.cc | 1828 bool LiteralBase::IsAllFirst() const { in IsAllFirst() function in xla::LiteralBase
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2269 !literal.IsAllFirst()))) { in HandleConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 1403 constant->literal().IsAllFirst()) { in HandleConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 792 literal.IsAllFirst()) { in ConstantLiteral()
|