Searched refs:IsAll (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 492 EXPECT_FALSE(tuple.IsAll(0)); in TEST_F()
493 EXPECT_FALSE(tuple.IsAll(1)); in TEST_F()
506 TEST_F(LiteralUtilTest, IsAll) { in TEST_F() argument
507 EXPECT_TRUE(LiteralUtil::CreateR0<bool>(false).IsAll(0)); in TEST_F()
508 EXPECT_TRUE(LiteralUtil::CreateR0<bool>(true).IsAll(1)); in TEST_F()
509 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(false).IsAll(1)); in TEST_F()
510 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(false).IsAll(2)); in TEST_F()
511 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(true).IsAll(0)); in TEST_F()
512 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(true).IsAll(2)); in TEST_F()
513 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(true).IsAll(-1)); in TEST_F()
[all …]
|
| D | literal.h | 207 bool IsAll(int8 value) const;
|
| D | literal.cc | 1723 bool LiteralBase::IsAll(int8 value) const { in IsAll() function in xla::LiteralBase
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 74 bool IsAll(const HloInstruction* op, int8 value) { in IsAll() function
77 return IsAll(op->operand(0), value); in IsAll()
79 return op->literal().IsAll(value); in IsAll()
807 if (IsAll(rhs, 0) && ReplaceInstructionIfSameShape(add, lhs)) { in HandleAdd()
812 if (IsAll(lhs, 0) && ReplaceInstructionIfSameShape(add, rhs)) { in HandleAdd()
1123 if (IsAll(rhs, 1) && ReplaceInstructionIfSameShape(logical_and, lhs)) { in HandleAnd()
1129 if (IsAll(lhs, 1) && ReplaceInstructionIfSameShape(logical_and, rhs)) { in HandleAnd()
1137 if (IsAll(rhs, 0) && ReplaceInstructionIfSameShape(logical_and, rhs)) { in HandleAnd()
1144 if (IsAll(lhs, 0) && ReplaceInstructionIfSameShape(logical_and, lhs)) { in HandleAnd()
1428 if (IsAll(rhs, 0) && ReplaceInstructionIfSameShape(sub, lhs)) { in HandleSubtract()
[all …]
|
| D | while_loop_simplifier.cc | 1011 if (!trip_counter && constant->literal().IsAll(1) && in TryMergeInductionVariables()
1013 while_init->operand(i)->literal().IsAll(0)) { in TryMergeInductionVariables()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | if_while_utils.cc | 52 all_values_are_static = literal.IsAll(0); in ConvertCompileTimeConstArgumentsToConst()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | op_types.h | 27 bool IsAll(const NodeDef& node);
|
| D | op_types.cc | 43 bool IsAll(const NodeDef& node) { return node.op() == "All"; } in IsAll() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 321 IsMin(node) || IsAll(node) || IsAny(node); in IsReduceOp()
|
| D | generic_layout_optimizer_transposer.cc | 2103 IsMin(node) || IsAll(node) || IsAny(node); in IsReduceOp()
|
| D | constant_folding.cc | 2817 } else if (IsAny(*node) || IsAll(*node)) { in ReplaceReductionWithIdentity()
|