| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| 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
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 221 const std::vector<HloInstruction*>& loop_state) { in ScatterLoopBody()
402 const std::vector<HloInstruction*>& loop_state) { in ExpandScatter()
|
| D | copy_insertion_test.cc | 418 auto loop_state = builder.AddInstruction( in BuildConditionComputation() local
439 auto loop_state = builder.AddInstruction( in BuildDependentBodyComputation() local
479 auto loop_state = builder.AddInstruction( in BuildDependentBodyComputation2() local
515 auto loop_state = builder.AddInstruction( in BuildDependentBodyOneReadOnlyComputation() local
552 auto loop_state = builder.AddInstruction( in BuildIndependentBodyComputation() local
603 auto loop_state = builder.AddInstruction(HloInstruction::CreateParameter( in BuildNestedBodyComputation() local
|
| D | gather_expander.cc | 352 const std::vector<HloInstruction*>& loop_state) { in ExpandInstruction()
|
| D | buffer_assignment_test.cc | 2027 auto loop_state = builder.AddInstruction( in BuildWhileBodyComputation() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 731 def _UpdatePendingAndEnqueueReady(grads, op, queue, pending_count, loop_state, argument
906 loop_state, argument
|
| D | rnn.py | 1187 loop_state): argument
|
| /external/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_nir_soa.c | 792 struct lp_build_loop_state loop_state; in emit_load_global() local
834 struct lp_build_loop_state loop_state; in emit_store_global() local
884 struct lp_build_loop_state loop_state; in emit_atomic_global() local
1046 struct lp_build_loop_state loop_state; in emit_load_mem() local
1128 struct lp_build_loop_state loop_state; in emit_store_mem() local
1183 struct lp_build_loop_state loop_state; in emit_atomic_mem() local
1778 struct lp_build_loop_state loop_state; in emit_vote() local
1888 struct lp_build_loop_state loop_state; in emit_load_scratch() local
1952 struct lp_build_loop_state loop_state; in emit_store_scratch() local
|
| D | lp_bld_format_soa.c | 1085 struct lp_build_loop_state loop_state; in lp_build_store_rgba_soa() local
|
| D | lp_bld_tgsi_soa.c | 3522 struct lp_build_loop_state loop_state; in load_emit() local
3642 struct lp_build_loop_state loop_state; in store_emit() local
3836 struct lp_build_loop_state loop_state; in atomic_emit() local
|
| D | lp_bld_sample_soa.c | 4074 struct lp_build_loop_state loop_state; in lp_build_do_atomic_soa() local
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_divergence_analysis.c | 829 struct divergence_state loop_state = *state; in visit_loop() local
|
| /external/mesa3d/src/gallium/drivers/r300/compiler/ |
| D | radeon_compiler.h | 74 struct emulate_loop_state loop_state; member
|
| /external/mesa3d/src/compiler/glsl/ |
| D | loop_analysis.cpp | 293 loop_state::loop_state() in loop_state() function in loop_state
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 2183 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2220 def loop_fn(time_, cell_output, cell_state, loop_state): argument
|
| /external/mesa3d/src/gallium/drivers/llvmpipe/ |
| D | lp_state_cs.c | 181 struct lp_build_loop_state loop_state[4]; in generate_compute() local
|
| D | lp_state_fs.c | 424 struct lp_build_for_loop_state *loop_state; member
592 struct lp_build_for_loop_state loop_state, sample_loop_state; in generate_fs_loop() local
|
| /external/mesa3d/src/gallium/auxiliary/draw/ |
| D | draw_llvm.c | 3391 struct lp_build_loop_state loop_state[2]; in draw_tcs_llvm_generate() local
|