Home
last modified time | relevance | path

Searched defs:parallel_iterations (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops.py40 def for_loop(loop_fn, loop_fn_dtypes, iters, parallel_iterations=None): argument
111 def pfor(loop_fn, iters, parallel_iterations=None): argument
183 def _pfor_impl(loop_fn, iters, parallel_iterations=None, pfor_config=None): argument
Dgradients.py28 def jacobian(output, inputs, use_pfor=True, parallel_iterations=None): argument
81 def batch_jacobian(output, inp, use_pfor=True, parallel_iterations=None): argument
Dtest_util.py54 parallel_iterations=None): argument
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/
Drnn.py47 parallel_iterations=None, argument
286 parallel_iterations=None, argument
/external/tensorflow/tensorflow/python/autograph/lang/
Ddirectives.py49 parallel_iterations=UNSPECIFIED, argument
/external/tensorflow/tensorflow/python/ops/
Drnn.py357 dtype=None, parallel_iterations=None, argument
492 dtype=None, parallel_iterations=None, swap_memory=False, argument
686 parallel_iterations, argument
903 parallel_iterations=None, swap_memory=False, scope=None): argument
Dfunctional_ops.py47 def foldl(fn, elems, initializer=None, parallel_iterations=10, back_prop=True, argument
156 def foldr(fn, elems, initializer=None, parallel_iterations=10, back_prop=True, argument
267 def scan(fn, elems, initializer=None, parallel_iterations=10, back_prop=True, argument
Dmap_fn.py38 def map_fn(fn, elems, dtype=None, parallel_iterations=None, back_prop=True, argument
Dcontrol_flow_ops.py212 parallel_iterations=10, argument
2167 parallel_iterations=10, argument
2197 def _init_from_args(self, maximum_iterations, parallel_iterations, back_prop, argument
2292 def parallel_iterations(self): member in WhileContext
3046 parallel_iterations=10, argument
3225 parallel_iterations=10, argument
Dwhile_v2.py60 parallel_iterations=10, argument
/external/tensorflow/tensorflow/contrib/rnn/python/ops/
Drnn.py132 parallel_iterations=None, argument
/external/tensorflow/tensorflow/core/protobuf/
Dcontrol_flow.proto57 int32 parallel_iterations = 2; field
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op.cc56 const string& body_fn_name, int parallel_iterations, in Run()
154 int parallel_iterations, Graph* graph, in LowerWhileHelper()
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Ddynamic_rnn_estimator.py218 parallel_iterations=32, argument
399 parallel_iterations=None, argument
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_map_ops.py40 parallel_iterations=None, argument
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Ddecoder.py159 parallel_iterations=32, argument
272 parallel_iterations=32, argument
/external/tensorflow/tensorflow/python/eager/
Dbackprop.py969 parallel_iterations=None, argument
1070 parallel_iterations=None, argument
/external/tensorflow/tensorflow/core/debug/
Ddebug_graph_utils.cc227 const AttrValue* parallel_iterations = in DeparallelizeWhileLoops() local
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc397 const int parallel_iterations, Status* status) { in AddControlEnter()
502 int parallel_iterations; in AddControlLoop() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer_test.cc46 AttrValue parallel_iterations; in AddEnterNode() local
Dloop_optimizer.cc244 AttrValue parallel_iterations; in HandleInvariantNode() local
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py3066 parallel_iterations=None, argument