Searched refs:IsAll (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 496 EXPECT_FALSE(tuple.IsAll(0)); in TEST_F()
497 EXPECT_FALSE(tuple.IsAll(1)); in TEST_F()
510 TEST_F(LiteralUtilTest, IsAll) { in TEST_F() argument
511 EXPECT_TRUE(LiteralUtil::CreateR0<bool>(false).IsAll(0)); in TEST_F()
512 EXPECT_TRUE(LiteralUtil::CreateR0<bool>(true).IsAll(1)); in TEST_F()
513 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(false).IsAll(1)); in TEST_F()
514 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(false).IsAll(2)); in TEST_F()
515 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(true).IsAll(0)); in TEST_F()
516 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(true).IsAll(2)); in TEST_F()
517 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(true).IsAll(-1)); in TEST_F()
[all …]
|
| D | literal.h | 178 bool IsAll(int8 value) const;
|
| D | literal.cc | 1555 bool LiteralBase::IsAll(int8 value) const { in IsAll() function in xla::LiteralBase
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 68 bool IsAll(const HloInstruction* op, int8 value) { in IsAll() function
71 return IsAll(op->operand(0), value); in IsAll()
73 return op->literal().IsAll(value); in IsAll()
494 if (IsAll(rhs, 0) && ReplaceInstructionIfSameShape(add, lhs)) { in HandleAdd()
499 if (IsAll(lhs, 0) && ReplaceInstructionIfSameShape(add, rhs)) { in HandleAdd()
575 if (IsAll(rhs, 1) && ReplaceInstructionIfSameShape(logical_and, lhs)) { in HandleAnd()
581 if (IsAll(lhs, 1) && ReplaceInstructionIfSameShape(logical_and, rhs)) { in HandleAnd()
588 if (IsAll(rhs, 0) && ReplaceInstructionIfSameShape(logical_and, rhs)) { in HandleAnd()
595 if (IsAll(lhs, 0) && ReplaceInstructionIfSameShape(logical_and, lhs)) { in HandleAnd()
832 if (IsAll(rhs, 0) && ReplaceInstructionIfSameShape(sub, lhs)) { in HandleSubtract()
[all …]
|
| D | while_loop_simplifier.cc | 799 if (!trip_counter && constant->literal().IsAll(1) && in TryMergeInductionVariables()
801 while_init->operand(i)->literal().IsAll(0)) { in TryMergeInductionVariables()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | op_types.h | 27 bool IsAll(const NodeDef& node);
|
| D | op_types.cc | 39 bool IsAll(const NodeDef& node) { return node.op() == "All"; } in IsAll() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 313 IsMin(node) || IsAll(node) || IsAny(node); in IsReduceOp()
|