Home
last modified time | relevance | path

Searched refs:IsOnes (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/transforms/utils/
Dop_cat_helper.cc104 bool OpCatHelper::IsOnes(TFOp op) { in IsOnes() function in mlir::tfg::OpCatHelper
110 return !value_op && IsOnes(value_op); in IsOnes()
Dop_cat_helper.h35 bool IsOnes(TFOp op);
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.h102 bool IsOnes(const NodeDef& node) const;
Dconstant_folding.cc1782 bool ConstantFolding::IsOnes(const NodeDef& node) const { in IsOnes() function in tensorflow::grappler::ConstantFolding
1790 return values != nullptr && IsOnes(*values); in IsOnes()
2568 const bool is_all_true = IsOnes(*predicate_node); in SimplifySelect()
2628 (!is_add_or_sub && IsOnes(*delta_node))) { in RemoveRedundantVariableUpdates()
3003 const bool x_is_one = x_is_zero ? false : IsOnes(*x); in SimplifyArithmeticOperations()
3032 const bool y_is_one = y_is_zero ? false : IsOnes(*y); in SimplifyArithmeticOperations()
Darithmetic_optimizer.cc2590 if (IsConstant(*input) || !IsOnes(*ones)) return OkStatus(); in TrySimplify()
2660 bool IsOnes(const NodeDef& node) const { in IsOnes() function in tensorflow::grappler::__anon2c8062030111::ReplaceMulWithBroadcastByTile
/external/tensorflow/tensorflow/core/transforms/constant_folding/
Dpass.cc2188 const bool x_is_one = x_is_zero ? false : helper_.IsOnes(x); in matchAndRewrite()
2242 const bool y_is_one = helper_.IsOnes(y); in matchAndRewrite()
3414 bool is_all_true = this->helper_.IsOnes(condition_op); in matchAndRewrite()