| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/ |
| H A D | control_flow_test.py | 67 def set_state(loop_vars): argument
87 def set_state(loop_vars): argument
109 def set_state(loop_vars): argument
131 def set_state(loop_vars): argument
153 def set_state(loop_vars): argument
176 def set_state(loop_vars): argument
208 def set_state(loop_vars): argument
228 def set_state(loop_vars): argument
313 def set_state(loop_vars): argument
335 def set_state(loop_vars): argument
[all …]
|
| H A D | control_flow.py | 842 def reduce_body(loop_vars, iterate): argument
1258 def aug_test(*loop_vars):
1265 def aug_body(*loop_vars):
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | while_v2_indexed_slices_rewriter.py | 28 def rewrite_grad_indexed_slices(grads, body_grad_graph, loop_vars, argument
109 forward_input, loop_vars): argument
248 def _update_indexed_slices_param(graph, loop_vars, init_slices, input_slices, argument
|
| H A D | while_v2.py | 73 loop_vars, argument
468 def _build_while_op(loop_vars, cond_graph, body_graph, output_shapes, argument
1379 def _pack_sequence_as(loop_vars_signature, flat_orig_loop_vars, loop_vars): argument
1395 def _tensor_array_to_flow(loop_vars): argument
|
| H A D | control_flow_ops.py | 2249 def BuildLoop(self, pred, body, loop_vars, shape_invariants, argument
2349 loop_vars, argument
2530 loop_vars, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | executor_test.cc | 594 int loop_iters, int loop_vars, bool lower, in BM_WhileLoopHelper()
733 const int loop_vars = state.range(1); in BM_LoweredWhileLoop() local
753 const int loop_vars = state.range(1); in BM_LoweredWhileLoopWithTransfer() local
771 const int loop_vars = state.range(1); in BM_FunctionalWhileLoop() local
|
| /aosp_15_r20/external/icu/icu4c/source/python/icutools/databuilder/ |
| H A D | utils.py | 98 def format_repeated_request_command(request, cmd_template, loop_vars, common_vars): argument
|
| /aosp_15_r20/external/cronet/third_party/icu/source/python/icutools/databuilder/ |
| H A D | utils.py | 98 def format_repeated_request_command(request, cmd_template, loop_vars, common_vars): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | cholesky_expander.cc | 77 XlaBuilder* body_builder) -> StatusOr<std::vector<XlaOp>> { in CholeskyUnblocked()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
| H A D | convert_to_ssa.cpp | 114 auto loop_vars = addControlFlowLoadStores(body_block); in addLoopLoadStores() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/risc/ |
| H A D | risc_ops.py | 266 loop_vars, argument
|
| /aosp_15_r20/external/mesa3d/src/compiler/nir/ |
| H A D | nir_loop_analyze.c | 72 nir_loop_variable *loop_vars; member
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | loopnest.cpp | 2510 std::unordered_set<VarPtr> loop_vars; in compressBuffer() local
|