Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_util.cc81 } else if (IsLoopCond(node)) { in ExtractWhileLoopFrames()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h162 bool IsLoopCond() const { return class_ == NC_LOOP_COND; } in IsLoopCond() function
817 inline bool IsLoopCond(const Node* node) { return node->IsLoopCond(); } in IsLoopCond() function
Dcontrol_flow.cc65 if (IsLoopCond(node)) { in ValidateControlFlowInfo()
Dgraph_partition.cc654 if (IsLoopCond(node)) { in AddControlFlow()
/external/tensorflow/tensorflow/core/common_runtime/
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/core/grappler/
Dop_types.h112 bool IsLoopCond(const NodeDef& node);
Dop_types.cc363 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/compiler/jit/
Dmark_for_compilation_pass.cc1083 if (!maybe_loop_cond->IsLoopCond()) { in IsIdentityDrivingConstsInLoop()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1764 if (node.IsLoopCond()) { in CreateOperation()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc1159 if (node->IsSwitch() && src->IsLoopCond()) { in PlaceOpsOnTPU()