| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_util_test.py | 40 self.assertTrue(control_flow_util.IsSwitch(switch))
45 self.assertTrue(control_flow_util.IsSwitch(ref_switch))
47 self.assertFalse(control_flow_util.IsSwitch(test_ops.int_output().op))
112 if control_flow_util.IsSwitch(n):
117 self.assertTrue(control_flow_util.IsSwitch(n))
134 if control_flow_util.IsSwitch(n):
139 self.assertTrue(control_flow_util.IsSwitch(n))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_util.py | 92 def IsSwitch(op): function
114 if not IsSwitch(op):
152 if IsSwitch(op):
304 if IsSwitch(op):
|
| D | control_flow_state.py | 665 dead_branch = util.IsSwitch(op)
836 if not util.IsSwitch(op):
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_if_op_test.cc | 94 ASSERT_FALSE(op->IsSwitch()); in TEST()
110 if (op->IsSwitch()) { in TEST()
223 if (op->IsSwitch()) ++switch_count; in TEST()
|
| D | lower_while_op_test.cc | 94 ASSERT_FALSE(op->IsSwitch()); in TEST()
123 if (op->IsSwitch()) { in TEST()
440 ASSERT_FALSE(op->IsSwitch()); in TEST()
463 if (op->IsSwitch()) { in TEST()
|
| D | lower_case_op_test.cc | 98 ASSERT_FALSE(op->IsSwitch()); in TEST()
114 if (op->IsSwitch()) { in TEST()
259 if (op->IsSwitch()) ++switch_count; in TEST()
|
| D | immutable_executor_state.cc | 104 if (IsSwitch(n) || IsMerge(n) || IsEnter(n) || IsExit(n)) { in Initialize()
195 item->is_recv_or_switch = IsRecv(n) || IsSwitch(n); in Initialize()
|
| D | lower_functional_ops_test.cc | 283 ASSERT_FALSE(op->IsSwitch()); in TEST()
|
| D | function_utils.cc | 114 if (IsRecv(e->src()) || IsSwitch(e->src())) { in GetTheOnlyDataEdge()
|
| /external/libchrome/base/ |
| D | command_line.cc | 58 bool IsSwitch(const CommandLine::StringType& string, in IsSwitch() function
89 if (parse_switches && IsSwitch(arg, &switch_string, &switch_value)) { in AppendSwitchesAndArguments()
468 if (parse_switches && IsSwitch(arg, &switch_string, &switch_value)) { in GetArgumentsStringInternal()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 157 bool IsSwitch() const { return class_ == NC_SWITCH; } in IsSwitch() function
176 (IsSwitch() || IsMerge() || IsEnter() || IsExit() || in IsControlFlow()
812 inline bool IsSwitch(const Node* node) { return node->IsSwitch(); } in IsSwitch() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer_factory.cc | 133 if (IsSwitch(node)) { in GetTransposer()
|
| D | dependency_optimizer.cc | 83 if (IsSwitch(*input)) { in SafeToRemoveIdentity()
111 if (IsMerge(node) || IsSwitch(node) || ModifiesFrameInfo(node)) { in SafeToConvertToNoOp()
|
| D | model_pruner.cc | 44 } else if (IsSwitch(*input.node)) { // Node is driven by switch. in IsTrivialIdentity()
|
| D | generic_layout_optimizer_transposer.cc | 1892 DCHECK(IsSwitch(*node->node())); in TransposeNode()
2057 IsSwitch(node) || IsTernaryOp(node) || IsUnaryGrad(node) || in IsLayoutAgnosticOp()
2149 if (IsSwitch(*node_def)) { in GetDataFanoutPorts()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.cc | 284 if (e_cond->IsControlEdge() || !e_cond->dst()->IsSwitch()) continue; in FindLoopExitNodes()
301 if (n->IsSwitch()) { in GetLoopSwitchNode()
307 TF_RET_CHECK(loop_switch_node->IsSwitch()) in GetLoopSwitchNode()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_cond.cc | 538 return IsSwitch(e->src()) ? BranchType(e->src_output()) in ExtractBodies()
557 if (IsSwitch(src)) { in ExtractBodies()
631 if (IsSwitch(src)) { in ExtractBodies()
737 if (!IsSwitch(in)) { in ExtractBodies()
1143 if (IsSwitch(src)) { in StateAlongEdge()
1328 if (IsSwitch(dst)) TF_RETURN_IF_ERROR(RemoveRedundantSwitch(dst)); in DetermineStates()
1349 } else if (IsSwitch(src)) { in DetermineAncestorState()
1495 if (IsSwitch(n)) { in FunctionalizeInternal()
|
| D | functionalize_while.cc | 278 if (!edge->IsControlEdge() && IsSwitch(edge->dst()) && in FunctionalizeLoop()
346 if (edge->dst_input() == 0 && IsSwitch(edge->dst()) && in FunctionalizeLoop()
|
| D | resource_util.cc | 54 return (n->IsEnter() || n->IsExit() || n->IsSwitch() || n->IsMerge() || in IsControlFlowV1Node()
|
| D | graph_compiler.cc | 149 TF_RET_CHECK(!n->IsRecv() && !n->IsSend() && !n->IsSwitch()) in Compile()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view.cc | 71 return IsSwitch(*input_node); in IsIdentityConsumingSwitch()
624 bool from_is_switch = IsSwitch(*from_node); in SwapNodeNames()
631 bool to_is_switch = IsSwitch(*to_node); in SwapNodeNames()
774 bool to_node_is_switch = IsSwitch(*to_node); in UpdateFanoutsInternal()
959 if (!IsSwitch(*fanin.node)) { in GetControllingFaninToAdd()
1259 if (to_fanin_is_control && IsSwitch(*to_fanin_node)) { in UpdateFanin()
|
| D | op_types.h | 192 bool IsSwitch(const NodeDef& node);
|
| D | op_types.cc | 583 bool IsSwitch(const NodeDef& node) { in IsSwitch() function
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference.cc | 184 bool is_loop_invariant = node->IsSwitch() && in PropagateShapes()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | scheduler.cc | 127 if (IsSwitch(n)) { in ComputeAlap()
|