Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/autograph/converters/
Dcontrol_flow.py354 def _state_constructs(self, loop_state, reserved_symbols): argument
443 def _create_for_loop_early_stopping(self, loop_state, state_ssf, argument
471 def _create_for_loop_with_state(self, loop_state, state_ssf, state_ast_tuple, argument
/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/compiler/xla/service/
Dscatter_expander.cc221 const std::vector<HloInstruction*>& loop_state) { in ScatterLoopBody()
402 const std::vector<HloInstruction*>& loop_state) { in ExpandScatter()
Dcopy_insertion_test.cc418 auto loop_state = builder.AddInstruction( in BuildConditionComputation() local
439 auto loop_state = builder.AddInstruction( in BuildDependentBodyComputation() local
476 auto loop_state = builder.AddInstruction( in BuildDependentBodyComputation2() local
512 auto loop_state = builder.AddInstruction( in BuildDependentBodyOneReadOnlyComputation() local
546 auto loop_state = builder.AddInstruction( in BuildIndependentBodyComputation() local
597 auto loop_state = builder.AddInstruction(HloInstruction::CreateParameter( in BuildNestedBodyComputation() local
Dgather_expander.cc352 const std::vector<HloInstruction*>& loop_state) { in ExpandInstruction()
Dbuffer_assignment_test.cc1875 auto loop_state = builder.AddInstruction( in BuildWhileBodyComputation() local
/external/tensorflow/tensorflow/python/ops/
Dgradients_util.py793 def _UpdatePendingAndEnqueueReady(grads, op, queue, pending_count, loop_state, argument
962 loop_state, argument
Drnn.py1149 emit_ta, state, loop_state): argument
/external/mesa3d/src/gallium/drivers/r300/compiler/
Dradeon_compiler.h74 struct emulate_loop_state loop_state; member
/external/mesa3d/src/compiler/glsl/
Dloop_analysis.cpp253 loop_state::loop_state() in loop_state() function in loop_state
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py2175 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2212 def loop_fn(time_, cell_output, cell_state, loop_state): argument
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_state_fs.c321 struct lp_build_for_loop_state loop_state; in generate_fs_loop() local