Home
last modified time | relevance | path

Searched refs:IsLoopCond (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_propagation.cc60 if ((node->IsSwitch() && src->IsLoopCond()) || in UpdateDeviceFromInputs()
Dlower_while_op_test.cc97 ASSERT_FALSE(op->IsLoopCond()); in TEST()
443 ASSERT_FALSE(op->IsLoopCond()); in TEST()
Dlower_functional_ops_test.cc286 ASSERT_FALSE(op->IsLoopCond()); in TEST()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_util.cc72 } else if (IsLoopCond(node)) { in ExtractWhileLoopFrames()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h169 bool IsLoopCond() const { return class_ == NC_LOOP_COND; } in IsLoopCond() function
850 inline bool IsLoopCond(const Node* node) { return node->IsLoopCond(); } in IsLoopCond() function
Dcontrol_flow.cc65 if (IsLoopCond(node)) { in ValidateControlFlowInfo()
Dgraph_partition.cc653 if (IsLoopCond(node)) { in AddControlFlow()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h113 bool IsLoopCond(const NodeDef& node);
Dop_types.cc365 bool IsLoopCond(const NodeDef& node) { return node.op() == "LoopCond"; } in IsLoopCond() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc626 if (!IsMerge(*switch_input) || !IsLoopCond(*switch_predicate)) { in CheckForDeadFanout()
/external/tensorflow/tensorflow/core/ir/
Dtf_op_names.cc498 bool TFGraphDialect::IsLoopCond(TFOp op) const { in IsLoopCond() function in mlir::tfg::TFGraphDialect
Dtf_op_names.inc92 bool IsLoopCond(TFOp op) const;
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass.cc1176 if (!maybe_loop_cond->IsLoopCond()) { in IsIdentityDrivingConstsInLoop()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1898 if (node.IsLoopCond()) { in CreateOperation()