Home
last modified time | relevance | path

Searched refs:vectorized_map (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dxla_control_flow_ops_test.py73 return pfor_control_flow_ops.vectorized_map(compute, x)
86 return pfor_control_flow_ops.vectorized_map(compute, x)
98 output = pfor_control_flow_ops.vectorized_map(compute, inp)
Dcontrol_flow_ops_test.py138 result = pfor_control_flow_ops.vectorized_map(compute,
149 result = pfor_control_flow_ops.vectorized_map(compute, x)
162 a_plus_1 = pfor_control_flow_ops.vectorized_map(lambda a: a + 1, a)
166 a_plus_a = pfor_control_flow_ops.vectorized_map(sum, (a, a))
170 a_plus_b = pfor_control_flow_ops.vectorized_map(sum, (a, b))
180 c = pfor_control_flow_ops.vectorized_map(outer_product, a)
188 pfor_control_flow_ops.vectorized_map(
1688 v_log_prob = lambda x: pfor_control_flow_ops.vectorized_map(log_prob, x)
2183 pfor_control_flow_ops.vectorized_map(
2196 pfor_control_flow_ops.vectorized_map(
[all …]
Dcontrol_flow_ops.py424 def vectorized_map(fn, elems, fallback_to_while_loop=True): function
/external/tensorflow/tensorflow/python/keras/integration_test/
Dvectorized_map_test.py40 per_example_gradients = tf.vectorized_map(model_fn, (inputs, labels))
Dforwardprop_test.py97 return tf.vectorized_map(_wrapper, tf.range(total_size))
/external/tensorflow/tensorflow/python/ops/
Dstandard_ops.py111 from tensorflow.python.ops.parallel_for.control_flow_ops import vectorized_map
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_interop_test.py335 c = tf.vectorized_map(outer_product, a)
340 c = tf.vectorized_map(lambda x: x.T, a)
/external/tensorflow/tensorflow/python/eager/
Dforwardprop.py174 return control_flow_ops.vectorized_map(
Dforwardprop_test.py93 return control_flow_ops.vectorized_map(
107 return control_flow_ops.vectorized_map(jac_mul, tangent_batch)
169 return control_flow_ops.vectorized_map(_wrapper, math_ops.range(total_size))
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Ddiscretization.py267 control_flow_ops.vectorized_map(
/external/tensorflow/tensorflow/python/autograph/operators/
Dpy_builtins.py611 mapped = parallel_ops.vectorized_map(key, iterable)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.pbtxt1156 name: "vectorized_map"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2512 name: "vectorized_map"
/external/tensorflow/
DRELEASE.md1112 * Update `tf.vectorized_map` to support vectorizing `tf.while_loop` and TensorList operations.
1319 * `pfor/vectorized_map`: Added support for vectorizing 56 more ops.
1603 * Fix bug where `vectorized_map` failed on inputs with unknown static shape.