Home
last modified time | relevance | path

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

/external/mesa3d/src/compiler/nir/
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
Dnir_divergence_analysis.c1358 struct divergence_state loop_state = *state; in visit_loop() local
/external/tensorflow/tensorflow/compiler/xla/service/
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()
Dgather_expander.cc387 const std::vector<HloInstruction*>& loop_state) { in ExpandInstruction()
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
Dbuffer_assignment_test.cc2103 auto loop_state = builder.AddInstruction( in BuildWhileBodyComputation() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
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
/external/tensorflow/tensorflow/python/ops/
Dgradients_util.py757 def _UpdatePendingAndEnqueueReady(grads, op, queue, pending_count, loop_state, argument
940 loop_state, argument
Drnn.py1231 loop_state): argument
/external/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_nir_soa.c1153 struct lp_build_loop_state loop_state; in emit_atomic_global() local
2308 struct lp_build_loop_state loop_state; in emit_vote() local
2373 struct lp_build_loop_state loop_state; in emit_ballot() local
2397 struct lp_build_loop_state loop_state; in emit_elect() local
2446 struct lp_build_loop_state loop_state; in emit_shuffle() local
Dlp_bld_format_soa.c1106 struct lp_build_loop_state loop_state; in lp_build_store_rgba_soa() local
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
Dlp_bld_sample_soa.c2270 struct lp_build_for_loop_state loop_state; in lp_build_sample_aniso() local
4560 struct lp_build_loop_state loop_state; in lp_build_do_atomic_soa() local
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
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
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_state_fs.c478 struct lp_build_for_loop_state *loop_state; member
711 struct lp_build_for_loop_state loop_state, sample_loop_state = {0}; in generate_fs_loop() local
Dlp_state_cs.c573 struct lp_build_loop_state loop_state[2]; in generate_compute() local
/external/mesa3d/src/gallium/auxiliary/draw/
Ddraw_llvm.c2999 struct lp_build_loop_state loop_state[2]; in draw_tcs_llvm_generate() local