| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_test.cc | 499 EXPECT_FALSE(tuple.IsAll(0)); in TEST_F()
500 EXPECT_FALSE(tuple.IsAll(1)); in TEST_F()
513 TEST_F(LiteralUtilTest, IsAll) { in TEST_F() argument
514 EXPECT_TRUE(LiteralUtil::CreateR0<bool>(false).IsAll(0)); in TEST_F()
515 EXPECT_TRUE(LiteralUtil::CreateR0<bool>(true).IsAll(1)); in TEST_F()
516 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(false).IsAll(1)); in TEST_F()
517 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(false).IsAll(2)); in TEST_F()
518 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(true).IsAll(0)); in TEST_F()
519 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(true).IsAll(2)); in TEST_F()
520 EXPECT_FALSE(LiteralUtil::CreateR0<bool>(true).IsAll(-1)); in TEST_F()
[all …]
|
| D | literal.cc | 1933 bool Literal::Piece::IsAll(const Literal& scalar) const { in IsAll() function in xla::Literal::Piece
1991 bool LiteralBase::IsAll(const Literal& scalar) const { in IsAll() function in xla::LiteralBase
1992 return root_piece().IsAll(scalar); in IsAll()
1995 bool LiteralBase::IsAll(int8_t value) const { in IsAll() function in xla::LiteralBase
2044 return root_piece().IsAll(scalar); in IsAll()
2069 return root_piece().IsAll(scalar); in IsAllFloat()
2088 return root_piece().IsAll(scalar); in IsAllComplex()
2104 return IsAll(first.Reshape({}).ValueOrDie()); in IsAllFirst()
|
| D | literal.h | 215 bool IsAll(const Literal& scalar) const;
228 bool IsAll(int8_t value) const;
607 bool IsAll(const Literal& scalar) const;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 75 bool IsAll(const HloInstruction* op, int8_t value) { in IsAll() function
78 return IsAll(op->operand(0), value); in IsAll()
80 return op->literal().IsAll(value); in IsAll()
86 bool IsAll(const HloInstruction* op, const Literal& scalar) { in IsAll() function
90 return IsAll(op->operand(0), scalar); in IsAll()
92 return op->literal().IsAll(scalar); in IsAll()
689 if (IsAll(rhs, 0) && ReplaceInstructionIfCompatible(add, lhs)) { in HandleAdd()
694 if (IsAll(lhs, 0) && ReplaceInstructionIfCompatible(add, rhs)) { in HandleAdd()
1010 if (IsAll(rhs, 1) && ReplaceInstructionIfCompatible(logical_and, lhs)) { in HandleAnd()
1016 if (IsAll(lhs, 1) && ReplaceInstructionIfCompatible(logical_and, rhs)) { in HandleAnd()
[all …]
|
| D | while_loop_simplifier.cc | 1158 if (!trip_counter && constant->literal().IsAll(1) && in TryMergeInductionVariables()
1160 while_init->operand(i)->literal().IsAll(0)) { in TryMergeInductionVariables()
|
| D | hlo_instructions.cc | 1519 if (literal().IsAll(0)) { in OperandsToStringWithCanonicalNameMap()
1522 if (literal().IsAll(1)) { in OperandsToStringWithCanonicalNameMap()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | value_inference.h | 53 bool AllValid() { return mask_.IsAll(0); } in AllValid()
|
| D | value_inference.cc | 1049 all_operands_values_static &= operands[i].IsAll(0); in AnalyzeIsDynamic()
1065 bool all_values_static = operands[index].IsAll(0); in AnalyzeIsDynamic()
1072 bool all_values_static = operands[i].IsAll(0); in AnalyzeIsDynamic()
1086 operands, [](Literal& operand) { return !operand.IsAll(0); }); in AnalyzeIsDynamic()
1099 operands, [](Literal& operand) { return !operand.IsAll(0); }); in AnalyzeIsDynamic()
1793 if (mask.IsAll(1)) { in AnalyzeConstant()
1808 if (mask.IsAll(1)) { in AnalyzeConstant()
1823 if (mask.IsAll(1)) { in AnalyzeConstant()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | if_while_utils.cc | 58 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/ir/ |
| D | tf_op_names.cc | 37 bool TFGraphDialect::IsAll(TFOp op) const { in IsAll() function in mlir::tfg::TFGraphDialect
689 IsAny(op) || IsAll(op); in IsReduction()
|
| D | tf_op_names.inc | 5 bool IsAll(TFOp op) const;
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_control_flow_flattening_test.cc | 704 EXPECT_TRUE(hlo_bound->literal().IsAll(expected_bound)); in CheckWhileBound()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.cc | 325 if (reference.IsAll(0)) { in RunAndCompareInternal()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 2095 IsMin(node) || IsAll(node) || IsAny(node); in IsReduceOp()
|
| D | constant_folding.cc | 2867 } else if (IsAny(*node) || IsAll(*node)) { in ReplaceReductionWithIdentity()
|
| /external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| D | pass.cc | 2088 else if (dialect_->IsAny(op) || dialect_->IsAll(op)) in ReplaceReductionWithIdentity()
|