| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | gradients.py | 28 def jacobian(output, inputs, use_pfor=True, parallel_iterations=None): argument
83 def batch_jacobian(output, inp, use_pfor=True, parallel_iterations=None): argument
|
| D | control_flow_ops.py | 46 def for_loop(loop_fn, loop_fn_dtypes, iters, parallel_iterations=None): argument
138 def pfor(loop_fn, iters, fallback_to_while_loop=True, parallel_iterations=None): argument
269 parallel_iterations=None, argument
|
| D | test_util.py | 55 parallel_iterations=None, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | functional_ops.py | 53 parallel_iterations=10, argument
180 parallel_iterations=10, argument
248 parallel_iterations=10, argument
376 parallel_iterations=10, argument
444 parallel_iterations=10, argument
708 parallel_iterations=10, argument
|
| D | map_fn.py | 50 parallel_iterations=None, argument
634 parallel_iterations=None, argument
|
| D | rnn.py | 354 parallel_iterations=None, argument
510 parallel_iterations=None, argument
706 parallel_iterations, argument
921 parallel_iterations=None, argument
|
| D | control_flow_ops.py | 225 parallel_iterations=10, argument
1458 parallel_iterations=10, argument
1488 def _init_from_args(self, maximum_iterations, parallel_iterations, back_prop, argument
1583 def parallel_iterations(self): member in WhileContext
2369 parallel_iterations=10, argument
2550 parallel_iterations=10, argument
|
| D | while_v2.py | 64 parallel_iterations=10, argument
425 parallel_iterations, name, num_original_outputs): argument
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn.py | 50 parallel_iterations=None, argument
289 parallel_iterations=None, argument
|
| /external/tensorflow/tensorflow/python/autograph/lang/ |
| D | directives.py | 51 parallel_iterations=UNSPECIFIED, argument
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scaling_gradient_tape.py | 187 parallel_iterations=None, argument
197 parallel_iterations=None, argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_map_ops.py | 33 parallel_iterations=None, argument
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | control_flow.proto | 58 int32 parallel_iterations = 2; field
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 66 const NameAttrList& body_fn, int parallel_iterations, in Run()
179 int parallel_iterations, Graph* graph, in LowerWhileHelper()
|
| D | immutable_executor_state.h | 74 int32 parallel_iterations; member
|
| D | accumulate_n_optimizer.cc | 87 int parallel_iterations; in Run() local
|
| D | immutable_executor_state.cc | 168 int parallel_iterations; in Initialize() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 1094 parallel_iterations=None, argument
1210 parallel_iterations=None, argument
|
| D | def_function_test.py | 282 parallel_iterations=1, argument
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_graph_utils.cc | 260 const AttrValue* parallel_iterations = in DeparallelizeWhileLoops() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_extract_outside_compilation.cc | 152 uint64_t parallel_iterations, Location loc, in CloneEmptyWhile()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 424 const int parallel_iterations, Status* status) { in AddControlEnter()
529 int parallel_iterations; in AddControlLoop() local
|
| /external/tensorflow/tensorflow/python/ops/risc/ |
| D | risc_ops.py | 272 parallel_iterations=10, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 749 IntegerAttr parallel_iterations; in ParseEnterOp() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | loop_optimizer.cc | 242 AttrValue parallel_iterations; in HandleInvariantNode() local
|