| /external/tensorflow/tensorflow/python/autograph/converters/ |
| D | control_flow.py | 322 loop_state = []
341 loop_state.append(s)
342 loop_state = frozenset(loop_state)
346 undefined_lives = loop_state - defined_in
352 return loop_state, reserved_symbols, possibly_undefs
354 def _state_constructs(self, loop_state, reserved_symbols): argument
355 loop_state = tuple(loop_state)
357 self.ctx.namer.new_symbol(s.ssf(), reserved_symbols) for s in loop_state
361 for name, ssf in zip(loop_state, state_ssf)
365 state_ast_tuple = gast.Tuple([n.ast() for n in loop_state], None)
[all …]
|
| /external/mesa3d/src/compiler/glsl/ |
| D | loop_analysis.h | 34 extern class loop_state *
46 unroll_loops(exec_list *instructions, loop_state *ls,
219 class loop_state {
221 ~loop_state();
233 loop_state();
242 friend loop_state *analyze_loop_variables(exec_list *instructions);
|
| D | loop_analysis.cpp | 253 loop_state::loop_state() in loop_state() function in loop_state
262 loop_state::~loop_state() in ~loop_state()
270 loop_state::insert(ir_loop *ir) in insert()
282 loop_state::get(const ir_loop *ir) in get()
357 loop_analysis(loop_state *loops);
371 loop_state *loops;
382 loop_analysis::loop_analysis(loop_state *loops) in loop_analysis()
804 loop_state *
807 loop_state *loops = new loop_state; in analyze_loop_variables()
|
| D | loop_unroll.cpp | 34 loop_unroll_visitor(loop_state *state, in loop_unroll_visitor()
50 loop_state *state;
584 unroll_loops(exec_list *instructions, loop_state *ls, in unroll_loops()
|
| D | glsl_parser_extras.cpp | 2235 loop_state *ls = analyze_loop_variables(ir); in do_common_optimization()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 174 loop_state = control_flow_ops.MaybeCreateControlFlowState(
184 return reachable_to_ops, pending_count, loop_state
616 reachable_to_ops, pending_count, loop_state = _PendingCount(
644 if loop_state:
645 loop_exits = loop_state.ProcessUnusedLoopExits(pending_count, to_ops_set)
648 _SetGrad(grads, y, loop_state.ZerosLikeForExit(y))
656 if loop_state:
657 loop_state.EnterGradWhileContext(op, before=True)
658 out_grads = _AggregatedGrads(grads, op, gradient_uid, loop_state,
660 if loop_state:
[all …]
|
| D | rnn.py | 1093 loop_state = (init_loop_state if init_loop_state is not None
1149 emit_ta, state, loop_state): argument
1171 next_time, next_output, cell_state, loop_state)
1179 loop_state = loop_state if next_loop_state is None else next_loop_state
1203 emit_ta, next_state, loop_state)
1208 emit_ta, state, loop_state],
|
| D | control_flow_ops.py | 1311 loop_state = None
1314 if loop_state is None:
1315 loop_state = ControlFlowState()
1318 loop_state.AddWhileContext(op, between_op_list, between_ops)
1320 loop_state.AddWhileContext(op, between_op_list, between_ops)
1321 return loop_state
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | while_transformer_test.cc | 42 auto loop_state = builder.AddInstruction(HloInstruction::CreateParameter( in BuildConditionComputation() local
46 limit_const->shape(), loop_state, tuple_index)); in BuildConditionComputation()
58 auto loop_state = builder.AddInstruction(HloInstruction::CreateParameter( in BuildBodyComputation() local
63 induction_variable_shape_, loop_state, ind_var_tuple_index)); in BuildBodyComputation()
70 data_shape_, loop_state, data_tuple_index)); in BuildBodyComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 221 const std::vector<HloInstruction*>& loop_state) { in ScatterLoopBody() argument
224 CHECK_EQ(loop_state.size(), 3); in ScatterLoopBody()
225 HloInstruction* operand = loop_state[0]; in ScatterLoopBody()
226 HloInstruction* scatter_indices = loop_state[1]; in ScatterLoopBody()
227 HloInstruction* updates = loop_state[2]; in ScatterLoopBody()
402 const std::vector<HloInstruction*>& loop_state) { in ExpandScatter() argument
403 return ScatterLoopBody(scatter, induction_var, loop_state); in ExpandScatter()
|
| D | copy_insertion_test.cc | 418 auto loop_state = builder.AddInstruction( in BuildConditionComputation() local
422 limit_const->shape(), loop_state, 0)); in BuildConditionComputation()
439 auto loop_state = builder.AddInstruction( in BuildDependentBodyComputation() local
444 induction_variable_shape_, loop_state, 0)); in BuildDependentBodyComputation()
451 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyComputation()
476 auto loop_state = builder.AddInstruction( in BuildDependentBodyComputation2() local
482 induction_variable_shape_, loop_state, 0)); in BuildDependentBodyComputation2()
491 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildDependentBodyComputation2()
495 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 2)); in BuildDependentBodyComputation2()
512 auto loop_state = builder.AddInstruction( in BuildDependentBodyOneReadOnlyComputation() local
[all …]
|
| D | gather_expander.cc | 352 const std::vector<HloInstruction*>& loop_state) { in ExpandInstruction() argument
353 return GatherLoopBody(*gather_instr, indvar, loop_state); in ExpandInstruction()
|
| D | buffer_assignment_test.cc | 1875 auto loop_state = builder.AddInstruction( in BuildWhileBodyComputation() local
1878 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 0)); in BuildWhileBodyComputation()
1880 HloInstruction::CreateGetTupleElement(data_shape_, loop_state, 1)); in BuildWhileBodyComputation()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 2175 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2177 loop_state = constant_op.constant([0])
2180 loop_state = array_ops.stack([array_ops.squeeze(loop_state) + 1])
2191 loop_state)
2194 loop_state = r[-1]
2195 self.assertEqual([10], loop_state.eval())
2212 def loop_fn(time_, cell_output, cell_state, loop_state): argument
2214 loop_state = tensor_array_ops.TensorArray(
2219 loop_state = loop_state.write(0, 1)
2222 loop_state = loop_state.write(time_,
[all …]
|
| /external/mesa3d/src/gallium/drivers/r300/compiler/ |
| D | radeon_compiler.h | 74 struct emulate_loop_state loop_state; member
|
| D | radeon_emulate_loops.c | 487 struct emulate_loop_state * s = &c->loop_state; in rc_transform_loops()
520 struct emulate_loop_state * s = &c->loop_state; in rc_emulate_loops()
|
| /external/mesa3d/src/gallium/drivers/llvmpipe/ |
| D | lp_state_fs.c | 321 struct lp_build_for_loop_state loop_state; in generate_fs_loop() local
395 lp_build_for_loop_begin(&loop_state, gallivm, in generate_fs_loop()
402 &loop_state.counter, 1, "mask_ptr"); in generate_fs_loop()
432 lp_build_interp_soa_update_pos_dyn(interp, gallivm, loop_state.counter); in generate_fs_loop()
446 &z_fb, &s_fb, loop_state.counter); in generate_fs_loop()
462 NULL, NULL, NULL, loop_state.counter, in generate_fs_loop()
475 lp_build_interp_soa_update_inputs_dyn(interp, gallivm, loop_state.counter); in generate_fs_loop()
571 &z_fb, &s_fb, loop_state.counter); in generate_fs_loop()
588 NULL, NULL, NULL, loop_state.counter, in generate_fs_loop()
602 &mask, z_fb, s_fb, loop_state.counter, in generate_fs_loop()
[all …]
|