/external/tensorflow/tensorflow/python/saved_model/ |
H A D | main_op_impl.py | 22 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__.py | 107 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 D | control_flow_util_test.py | 23 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 D | control_flow_ops_py_test.py | 17 """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 D | numerics_test.py | 27 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 D | logging_ops_test.py | 24 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 D | absolute_value.py | 25 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 D | weibull.py | 24 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 D | multitask_optimizer_wrapper.py | 27 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 D | multiple_dispatch.py | 24 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 D | numerics.py | 26 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 D | resources.py | 28 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 D | confusion_matrix.py | 31 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 D | weights_broadcast_ops.py | 28 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 D | control_flow_ops_test.py | 15 """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 D | standard_ops.py | 46 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 D | control_flow_test.py | 24 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 D | losses.py | 23 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 D | random_tensor_pool_impl.py | 36 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 D | sampling_ops.py | 24 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 D | nullary_ops_test.py | 25 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 D | base_split_handler.py | 23 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 D | mvn_full_covariance.py | 25 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 D | fwd_gradients.py | 23 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 D | util_ops.py | 26 from tensorflow.python.ops import control_flow_ops namespace 66 a, b = control_flow_ops.while_loop(cond, body, [a, b], back_prop=False)
|