Searched refs:control_flow_ops (Results 1 - 25 of 275) sorted by relevance

1234567891011

/external/tensorflow/tensorflow/python/saved_model/
H A Dmain_op_impl.py22 from tensorflow.python.ops import control_flow_ops namespace
41 return control_flow_ops.group(init, init_local, init_tables)
60 main_op_with_restore = control_flow_ops.group(restore_op_name)
/external/tensorflow/tensorflow/contrib/framework/
H A D__init__.py107 from tensorflow.python.ops.control_flow_ops import smart_cond
108 from tensorflow.python.ops.control_flow_ops import smart_constant_value
/external/tensorflow/tensorflow/python/kernel_tests/
H A Dcontrol_flow_util_test.py23 from tensorflow.python.ops import control_flow_ops namespace
32 switch_false, _ = control_flow_ops.switch(1, True)
36 ref_switch_false, _ = control_flow_ops.ref_switch(test_ops.ref_output(),
61 exit_op = control_flow_ops.exit(1).op
64 ref_exit = control_flow_ops.exit(test_ops.ref_output()).op
H A Dcontrol_flow_ops_py_test.py17 """Tests for tensorflow.ops.control_flow_ops."""
43 from tensorflow.python.ops import control_flow_ops namespace
121 _, r_s = control_flow_ops.while_loop(
133 v = control_flow_ops._Identity(v)
135 v2 = control_flow_ops.with_dependencies([op], v)
145 enter_v = control_flow_ops._Enter(v, "foo_1", is_constant=True)
149 v2 = control_flow_ops.with_dependencies([op], enter_v)
150 v3 = control_flow_ops.exit(v2)
159 v1 = control_flow_ops._SwitchRefOrTensor(v._ref(), p) # pylint: disable=protected-access
171 exit_op = control_flow_ops
[all...]
H A Dnumerics_test.py27 from tensorflow.python.ops import control_flow_ops namespace
74 a = control_flow_ops.with_dependencies([check], a)
84 a = control_flow_ops.with_dependencies([check], a)
94 a = control_flow_ops.with_dependencies([check], a)
108 _ = control_flow_ops.cond(predicate,
120 _ = control_flow_ops.while_loop(lambda _: predicate,
H A Dlogging_ops_test.py24 from tensorflow.python.ops import control_flow_ops namespace
42 control_flow_ops.Assert(
52 control_flow_ops.Assert(
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
H A Dabsolute_value.py25 from tensorflow.python.ops import control_flow_ops namespace
99 event_ndims = control_flow_ops.with_dependencies(
115 y = control_flow_ops.with_dependencies(
127 zeros = control_flow_ops.with_dependencies(
H A Dweibull.py24 from tensorflow.python.ops import control_flow_ops namespace
80 self._scale = control_flow_ops.with_dependencies([
85 self._concentration = control_flow_ops.with_dependencies([
139 return control_flow_ops.with_dependencies([is_valid], x)
149 return control_flow_ops.with_dependencies([is_positive, less_than_one], y)
/external/tensorflow/tensorflow/contrib/opt/python/training/
H A Dmultitask_optimizer_wrapper.py27 from tensorflow.python.ops import control_flow_ops namespace
43 return control_flow_ops.cond(all_zeros, control_flow_ops.no_op,
131 return control_flow_ops.cond(
/external/tensorflow/tensorflow/contrib/py2tf/utils/
H A Dmultiple_dispatch.py24 from tensorflow.python.ops import control_flow_ops namespace
42 return control_flow_ops.cond(condition, true_fn, false_fn)
81 return control_flow_ops.while_loop(cond_fn, body_fn, init_args)
/external/tensorflow/tensorflow/python/ops/
H A Dnumerics.py26 from tensorflow.python.ops import control_flow_ops namespace
46 out = control_flow_ops.with_dependencies([verify_input], t)
100 return control_flow_ops.group(*check_op)
H A Dresources.py28 from tensorflow.python.ops import control_flow_ops namespace
116 return control_flow_ops.group(*[r.create for r in resource_list], name=name)
117 return control_flow_ops.no_op(name=name)
H A Dconfusion_matrix.py31 from tensorflow.python.ops import control_flow_ops namespace
84 predictions = control_flow_ops.cond(
90 labels = control_flow_ops.cond(
160 labels = control_flow_ops.with_dependencies(
164 predictions = control_flow_ops.with_dependencies(
174 labels = control_flow_ops.with_dependencies(
178 predictions = control_flow_ops.with_dependencies(
H A Dweights_broadcast_ops.py28 from tensorflow.python.ops import control_flow_ops namespace
53 return control_flow_ops.cond(
97 return control_flow_ops.no_op(name="static_scalar_check_success")
117 return control_flow_ops.no_op(name="static_dims_check_success")
127 is_valid_shape = control_flow_ops.cond(
133 return control_flow_ops.Assert(is_valid_shape, data, name=scope)
H A Dcontrol_flow_ops_test.py15 """Tests for control_flow_ops.py."""
35 from tensorflow.python.ops import control_flow_ops namespace
72 control_flow_ops.group(a.op, b.op, c.op, name="root")
86 control_flow_ops.group(a.op, b.op, name="root")
103 control_flow_ops.group(a.op, b.op, c.op, d.op, name="root")
122 control_flow_ops.group([a.op, b.op], name="root")
133 control_flow_ops.group(1, 2)
144 control_flow_ops.with_dependencies(
156 const_with_dep = control_flow_ops.with_dependencies(
170 const_with_dep = control_flow_ops
[all...]
H A Dstandard_ops.py46 from tensorflow.python.ops.control_flow_ops import Assert
47 from tensorflow.python.ops.control_flow_ops import case
48 from tensorflow.python.ops.control_flow_ops import cond
49 from tensorflow.python.ops.control_flow_ops import group
50 from tensorflow.python.ops.control_flow_ops import no_op
51 from tensorflow.python.ops.control_flow_ops import tuple # pylint: disable=redefined-builtin
53 from tensorflow.python.ops.control_flow_ops import while_loop
103 from tensorflow.python.ops import control_flow_ops as _control_flow_ops
/external/tensorflow/tensorflow/contrib/py2tf/converters/
H A Dcontrol_flow_test.py24 from tensorflow.python.ops import control_flow_ops namespace
43 with self.compiled(node, control_flow_ops.while_loop) as result:
58 with self.compiled(node, control_flow_ops.while_loop) as result:
76 with self.compiled(node, control_flow_ops.cond) as result:
93 with self.compiled(node, control_flow_ops.cond) as result:
/external/tensorflow/tensorflow/contrib/boosted_trees/python/utils/
H A Dlosses.py23 from tensorflow.python.ops import control_flow_ops namespace
43 return unweighted_loss * weights, control_flow_ops.no_op()
105 return unweighted_loss, control_flow_ops.no_op()
107 return unweighted_loss * weights, control_flow_ops.no_op()
125 return unweighted_loss * weights, control_flow_ops.no_op()
213 return unweighted_loss * weights, control_flow_ops.no_op()
/external/tensorflow/tensorflow/contrib/gan/python/features/python/
H A Drandom_tensor_pool_impl.py36 from tensorflow.python.ops import control_flow_ops namespace
124 return control_flow_ops.cond(prob, lambda: dequeue_values,
127 output_values = _to_tuple(control_flow_ops.cond(
/external/tensorflow/tensorflow/contrib/training/python/training/
H A Dsampling_ops.py24 from tensorflow.python.ops import control_flow_ops namespace
115 control_flow_ops.with_dependencies([
217 assert_op = control_flow_ops.Assert(
225 init_probs = control_flow_ops.with_dependencies([assert_op], init_probs)
230 accept_probs = control_flow_ops.cond(
325 checked_probs = control_flow_ops.with_dependencies([
354 labels = control_flow_ops.with_dependencies([lbl_assert], labels)
356 control_flow_ops.with_dependencies([
363 labels = control_flow_ops.with_dependencies([
/external/tensorflow/tensorflow/compiler/tests/
H A Dnullary_ops_test.py25 from tensorflow.python.ops import control_flow_ops namespace
41 output = control_flow_ops.no_op()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/
H A Dbase_split_handler.py23 from tensorflow.python.ops import control_flow_ops namespace
128 return control_flow_ops.group(update_1, *update_2[self])
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
H A Dmvn_full_covariance.py25 from tensorflow.python.ops import control_flow_ops namespace
170 covariance_matrix = control_flow_ops.with_dependencies([
/external/tensorflow/tensorflow/contrib/nn/python/ops/
H A Dfwd_gradients.py23 from tensorflow.python.ops import control_flow_ops namespace
65 assert_unused = control_flow_ops.Assert(False, [1], name='fwd_gradients')
/external/tensorflow/tensorflow/contrib/signal/python/ops/
H A Dutil_ops.py26 from tensorflow.python.ops import control_flow_ops namespace
66 a, b = control_flow_ops.while_loop(cond, body, [a, b], back_prop=False)

Completed in 567 milliseconds

1234567891011