| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | functional_ops.py | 49 parallel_iterations=10, argument
177 parallel_iterations=10, argument
245 parallel_iterations=10, argument
374 parallel_iterations=10, argument
442 parallel_iterations=10, argument
707 parallel_iterations=10, argument
|
| H A D | rnn.py | 353 parallel_iterations=None, argument
509 parallel_iterations=None, argument
768 parallel_iterations, argument
985 parallel_iterations=None, argument
|
| H A D | control_flow_ops.py | 223 parallel_iterations=10, argument
1416 parallel_iterations=10, argument
1446 def _init_from_args(self, maximum_iterations, parallel_iterations, back_prop, argument
1541 def parallel_iterations(self): member in WhileContext
2351 parallel_iterations=10, argument
2532 parallel_iterations=10, argument
|
| H A D | map_fn.py | 47 parallel_iterations=None, argument
631 parallel_iterations=None, argument
|
| H A D | while_v2.py | 75 parallel_iterations=10, argument
469 parallel_iterations, name, num_original_outputs, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ |
| H A D | control_flow_ops.py | 45 def for_loop(loop_fn, loop_fn_dtypes, iters, parallel_iterations=None): argument
141 parallel_iterations=None, argument
276 parallel_iterations=None, argument
|
| H A D | gradients.py | 24 def jacobian(output, inputs, use_pfor=True, parallel_iterations=None): argument
79 def batch_jacobian(output, inp, use_pfor=True, parallel_iterations=None): argument
|
| H A D | test_util.py | 51 parallel_iterations=None, argument
|
| H A D | control_flow_ops_test.py | 2527 def test_create_composite_inside_loop(self, parallel_iterations): argument
2542 self, parallel_iterations): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | lower_while_op.cc | 66 const NameAttrList& body_fn, int parallel_iterations, in Run()
182 int parallel_iterations, Graph* graph, in LowerWhileHelper()
|
| H A D | immutable_executor_state.h | 74 int32 parallel_iterations; member
|
| H A D | accumulate_n_optimizer.cc | 87 int parallel_iterations; in Run() local
|
| H A D | immutable_executor_state.cc | 169 int parallel_iterations; in Initialize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | backprop.py | 1135 parallel_iterations=None, argument
1248 parallel_iterations=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/lang/ |
| H A D | directives.py | 47 parallel_iterations=UNSPECIFIED, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_map_ops.py | 30 parallel_iterations=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
| H A D | control_flow.proto | 58 int32 parallel_iterations = 2; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph_partition.cc | 423 const int parallel_iterations, Status* status) { in AddControlEnter()
528 int parallel_iterations; in AddControlLoop() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
| H A D | debug_graph_utils.cc | 260 const AttrValue* parallel_iterations = in DeparallelizeWhileLoops() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | tf_to_tfrt.cc | 1253 int64_t parallel_iterations = in matchAndRewrite() local
1374 int64_t parallel_iterations = in GetWhileBodyFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/risc/ |
| H A D | risc_ops.py | 268 parallel_iterations=10, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_executor.cc | 734 IntegerAttr parallel_iterations; in parse() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_extract_outside_compilation.cc | 224 uint64_t parallel_iterations, Location loc, in CloneEmptyWhile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | loop_optimizer.cc | 242 AttrValue parallel_iterations; in HandleInvariantNode() local
|
| H A D | loop_optimizer_test.cc | 48 AttrValue parallel_iterations; in AddEnterNode() local
|