/aosp_15_r20/external/mesa3d/src/compiler/nir/ |
H A D | nir_opt_dce.c | 114 struct loop_state { struct 120 dce_block(nir_block *block, BITSET_WORD *defs_live, struct loop_state *loop, in dce_block() argument
|
H A D | nir_divergence_analysis.c | 1206 struct divergence_state loop_state = *state; in visit_loop() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | scatter_expander.cc | 250 absl::Span<HloInstruction* const> loop_state) { in ScatterLoopBody() 467 std::vector<HloInstruction*> loop_state; in ExpandInstruction() local 476 const std::vector<HloInstruction*>& loop_state) { in ExpandInstruction()
|
H A D | copy_insertion_test.cc | 347 auto loop_state = builder.AddInstruction( in BuildConditionComputation() local 368 auto loop_state = builder.AddInstruction( in BuildDependentBodyComputation() local 408 auto loop_state = builder.AddInstruction( in BuildDependentBodyComputation2() local 444 auto loop_state = builder.AddInstruction( in BuildDependentBodyOneReadOnlyComputation() local 481 auto loop_state = builder.AddInstruction( in BuildIndependentBodyComputation() local 532 auto loop_state = builder.AddInstruction(HloInstruction::CreateParameter( in BuildNestedBodyComputation() local
|
H A D | gather_expander.cc | 387 const std::vector<HloInstruction*>& loop_state) { in ExpandInstruction()
|
H A D | buffer_assignment_test.cc | 2103 auto loop_state = builder.AddInstruction( in BuildWhileBodyComputation() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | while_transformer_test.cc | 42 auto loop_state = builder.AddInstruction(HloInstruction::CreateParameter( in BuildConditionComputation() local 58 auto loop_state = builder.AddInstruction(HloInstruction::CreateParameter( in BuildBodyComputation() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | gradients_util.py | 757 def _UpdatePendingAndEnqueueReady(grads, op, queue, pending_count, loop_state, argument 940 loop_state, argument
|
H A D | rnn.py | 1231 loop_state): argument
|
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/gallivm/ |
H A D | lp_bld_nir_soa.c | 1116 struct lp_build_loop_state loop_state; in emit_atomic_global() local 2271 struct lp_build_loop_state loop_state; in emit_vote() local 2336 struct lp_build_loop_state loop_state; in emit_ballot() local 2360 struct lp_build_loop_state loop_state; in emit_elect() local 2409 struct lp_build_loop_state loop_state; in emit_shuffle() local 2440 struct lp_build_loop_state loop_state; in emit_reduce() local
|
H A D | lp_bld_format_soa.c | 1106 struct lp_build_loop_state loop_state; in lp_build_store_rgba_soa() local
|
H A D | lp_bld_tgsi_soa.c | 3547 struct lp_build_loop_state loop_state; in load_emit() local 3669 struct lp_build_loop_state loop_state; in store_emit() local 3864 struct lp_build_loop_state loop_state; in atomic_emit() local
|
H A D | lp_bld_sample_soa.c | 4891 struct lp_build_loop_state loop_state; in lp_build_do_atomic_soa() local
|
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
H A D | rnn_cell_test.py | 2263 def loop_fn(time_, cell_output, cell_state, loop_state): argument 2300 def loop_fn(time_, cell_output, cell_state, loop_state): argument
|
/aosp_15_r20/external/mesa3d/src/gallium/drivers/llvmpipe/ |
H A D | lp_state_fs.c | 439 struct lp_build_for_loop_state *loop_state; member 672 struct lp_build_for_loop_state loop_state, sample_loop_state = {0}; in generate_fs_loop() local
|
H A D | lp_state_cs.c | 576 struct lp_build_loop_state loop_state[2]; in generate_compute() local
|
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/draw/ |
H A D | draw_llvm.c | 3007 struct lp_build_loop_state loop_state[2]; in draw_tcs_llvm_generate() local
|