Home
last modified time | relevance | path

Searched refs:smart_cond (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
Dsmart_cond_test.py24 from tensorflow.python.framework import smart_cond
44 z = smart_cond.smart_cond(True, lambda: math_ops.multiply(x, 16),
54 z = smart_cond.smart_cond(False, lambda: math_ops.multiply(x, 16),
62 y = smart_cond.smart_cond(x > 0, lambda: constant_op.constant(1),
74 z = smart_cond.smart_cond(x * y > 0, lambda: constant_op.constant(1),
82 y = smart_cond.smart_cond(x > 0, lambda: constant_op.constant(1),
92 smart_cond.smart_cond(True, false_fn=lambda: x)
99 smart_cond.smart_cond(True, lambda: x)
109 y = smart_cond.smart_case(conditions, default=raise_exception,
111 z = smart_cond.smart_case(conditions, default=raise_exception,
[all …]
Dsmart_cond.py27 def smart_cond(pred, true_fn=None, false_fn=None, name=None): function
118 smart_cond, pred_fn_pairs, default, exclusive, name,
/external/tensorflow/tensorflow/contrib/framework/
D__init__.py115 from tensorflow.python.framework.smart_cond import smart_case
116 from tensorflow.python.framework.smart_cond import smart_cond
117 from tensorflow.python.framework.smart_cond import smart_constant_value
DBUILD74 "//tensorflow/python:smart_cond",
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dutils_test.py113 o = utils.smart_cond(constant_op.constant(v), fn1, fn2)
121 o = utils.smart_cond(constant_op.constant(v), fn1, fn2)
130 o = utils.smart_cond(constant_op.constant(v), fn1, fn2)
140 o = utils.smart_cond(constant_op.constant(v), fn1, fn2)
153 o = utils.smart_cond(p, fn1, fn2)
163 o = utils.smart_cond(p, fn1, fn2)
173 o = utils.smart_cond(p, fn1, fn2)
184 o = utils.smart_cond(p, fn1, fn2)
Dutils.py197 def smart_cond(pred, fn1, fn2, name=None): function
Dlayers.py382 outputs, mean, variance = utils.smart_cond(
404 outputs = utils.smart_cond(is_training, _force_updates, no_updates)
416 update_mean, update_variance = utils.smart_cond(
802 mean, variance = utils.smart_cond(is_training, _force_updates,
814 update_mean, update_variance = utils.smart_cond(
820 mean, variance = utils.smart_cond(is_training, vars_fn, moving_vars_fn)
/external/tensorflow/tensorflow/examples/saved_model/integration_tests/
Dutil.py25 from tensorflow.python.framework import smart_cond
99 result = smart_cond.smart_cond(training,
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization.py462 output, mean, variance = tf_utils.smart_cond(
475 momentum = tf_utils.smart_cond(training,
525 r = tf_utils.smart_cond(training, lambda: r, lambda: array_ops.ones_like(r))
526 d = tf_utils.smart_cond(training,
553 return tf_utils.smart_cond(training, _do_update, _fake_update)
628 adj_scale = tf_utils.smart_cond(training,
631 adj_bias = tf_utils.smart_cond(training,
647 mean = tf_utils.smart_cond(training,
650 variance = tf_utils.smart_cond(training,
688 mean_update = tf_utils.smart_cond(
[all …]
Dcore.py159 output = tf_utils.smart_cond(training,
/external/tensorflow/tensorflow/python/layers/
Dutils.py24 from tensorflow.python.framework import smart_cond as smart_module
179 def smart_cond(pred, true_fn=None, false_fn=None, name=None): function
201 return smart_module.smart_cond(
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Ddistribution_util.py23 from tensorflow.python.framework import smart_cond
558 x, perm=smart_cond.smart_cond(source_idx < dest_idx,
564 return smart_cond.smart_cond(math_ops.equal(source_idx, dest_idx),
/external/tensorflow/tensorflow/python/keras/
Dlosses.py26 from tensorflow.python.framework import smart_cond
725 updated_y_true = smart_cond.smart_cond(is_binary,
860 y_true = smart_cond.smart_cond(label_smoothing,
882 y_true = smart_cond.smart_cond(label_smoothing,
/external/tensorflow/tensorflow/python/keras/utils/
Dtf_utils.py25 from tensorflow.python.framework import smart_cond as smart_module
33 def smart_cond(pred, true_fn=None, false_fn=None, name=None): function
55 return smart_module.smart_cond(
/external/tensorflow/tensorflow/contrib/crf/python/ops/
Dcrf.py110 return utils.smart_cond(
162 return utils.smart_cond(
221 return utils.smart_cond(
592 return utils.smart_cond(
/external/tensorflow/tensorflow/python/ops/
Dsummary_ops_v2.py41 from tensorflow.python.framework import smart_cond
644 return smart_cond.smart_cond(
673 op = smart_cond.smart_cond(
/external/tensorflow/tensorflow/contrib/eager/python/
Dmetrics_impl.py27 from tensorflow.python.framework import smart_cond
358 smart_cond.smart_cond(write_summary,
/external/tensorflow/tensorflow/contrib/quantize/python/
Dfold_batch_norms.py410 bn_decay_mean_out = utils.smart_cond(
422 bn_decay_var_out = utils.smart_cond(
432 correction_recip = utils.smart_cond(
438 correction_offset = utils.smart_cond(
/external/tensorflow/tensorflow/python/training/
Dinput.py530 return utils.smart_cond(
536 out_tensor = utils.smart_cond(
732 enqueue_ops = [utils.smart_cond(
749 enqueue_ops = [utils.smart_cond(
/external/tensorflow/tensorflow/python/data/ops/
DBUILD23 "//tensorflow/python:smart_cond",
Ddataset_ops.py48 from tensorflow.python.framework import smart_cond
2943 tensor_util.constant_value(self._batch_size) if smart_cond.
2955 if smart_cond.smart_constant_value(self._drop_remainder) is False:
/external/tensorflow/tensorflow/contrib/training/python/training/
Dbucket_ops.py242 maybe_enqueue = utils.smart_cond(
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
Dgraph_io.py315 keys, examples_proto = utils.smart_cond(
/external/tensorflow/tensorflow/python/
DBUILD1048 name = "smart_cond",
1049 srcs = ["framework/smart_cond.py"],
1067 ":smart_cond",
3342 ":smart_cond",
5415 ":smart_cond",
5434 ":smart_cond",
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_test.py210 output = tf_utils.smart_cond(

12