Home
last modified time | relevance | path

Searched defs:loop_state (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/mesa3d/src/compiler/nir/
H A Dnir_opt_dce.c114 struct loop_state { struct
120 dce_block(nir_block *block, BITSET_WORD *defs_live, struct loop_state *loop, in dce_block() argument
H A Dnir_divergence_analysis.c1206 struct divergence_state loop_state = *state; in visit_loop() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dscatter_expander.cc250 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 Dcopy_insertion_test.cc347 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 Dgather_expander.cc387 const std::vector<HloInstruction*>& loop_state) { in ExpandInstruction()
H A Dbuffer_assignment_test.cc2103 auto loop_state = builder.AddInstruction( in BuildWhileBodyComputation() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dwhile_transformer_test.cc42 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 Dgradients_util.py757 def _UpdatePendingAndEnqueueReady(grads, op, queue, pending_count, loop_state, argument
940 loop_state, argument
H A Drnn.py1231 loop_state): argument
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/gallivm/
H A Dlp_bld_nir_soa.c1116 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 Dlp_bld_format_soa.c1106 struct lp_build_loop_state loop_state; in lp_build_store_rgba_soa() local
H A Dlp_bld_tgsi_soa.c3547 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 Dlp_bld_sample_soa.c4891 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 Drnn_cell_test.py2263 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 Dlp_state_fs.c439 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 Dlp_state_cs.c576 struct lp_build_loop_state loop_state[2]; in generate_compute() local
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/draw/
H A Ddraw_llvm.c3007 struct lp_build_loop_state loop_state[2]; in draw_tcs_llvm_generate() local