| /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
181 parallel_iterations=10, argument
249 parallel_iterations=10, argument
378 parallel_iterations=10, argument
446 parallel_iterations=10, argument
711 parallel_iterations=10, argument
|
| D | rnn.py | 354 parallel_iterations=None, argument
510 parallel_iterations=None, argument
767 parallel_iterations, argument
982 parallel_iterations=None, argument
|
| D | map_fn.py | 50 parallel_iterations=None, argument
634 parallel_iterations=None, argument
|
| D | control_flow_ops.py | 227 parallel_iterations=10, argument
1472 parallel_iterations=10, argument
1502 def _init_from_args(self, maximum_iterations, parallel_iterations, back_prop, argument
1597 def parallel_iterations(self): member in WhileContext
2386 parallel_iterations=10, argument
2567 parallel_iterations=10, argument
|
| D | while_v2.py | 78 parallel_iterations=10, argument
457 parallel_iterations, name, num_original_outputs, 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/autograph/lang/ |
| D | directives.py | 51 parallel_iterations=UNSPECIFIED, 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()
182 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 | 1104 parallel_iterations=None, argument
1220 parallel_iterations=None, argument
|
| D | def_function_test.py | 345 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/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/core/grappler/optimizers/ |
| D | loop_optimizer.cc | 242 AttrValue parallel_iterations; in HandleInvariantNode() local
|
| D | loop_optimizer_test.cc | 48 AttrValue parallel_iterations; in AddEnterNode() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_extract_outside_compilation.cc | 224 uint64_t parallel_iterations, Location loc, in CloneEmptyWhile()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 749 IntegerAttr parallel_iterations; in ParseEnterOp() local
|