Searched refs:induction_var (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | deadness_analysis_test.cc | 86 Output induction_var; member
198 Output induction_var; member
569 Output iv0 = CreateInductionVariable(root, "iv0", "fr0", 0).induction_var; in TEST()
570 Output iv1 = CreateInductionVariable(root, "iv1", "fr0", 0).induction_var; in TEST()
571 Output iv2 = CreateInductionVariable(root, "iv2", "fr0", 1).induction_var; in TEST()
621 .induction_var; in TEST()
624 .induction_var; in TEST()
643 EXPECT_EQ(predicate_map[ControlOutputFor(iv.induction_var)], in TEST()
646 predicate_map[ControlOutputFor(iv.induction_var)]); in TEST()
648 predicate_map[ControlOutputFor(iv.induction_var)]); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 220 HloInstruction* scatter, HloInstruction* induction_var, in ScatterLoopBody() argument
233 MakeBroadcastHlo(induction_var, /*broadcast_dimensions=*/{}, in ScatterLoopBody()
411 [&](HloInstruction* induction_var, in ExpandInstruction()
413 return ScatterLoopBody(scatter, induction_var, loop_state); in ExpandInstruction()
|
| D | gather_expander.cc | 151 const HloInstruction& gather, HloInstruction* induction_var, in GatherLoopBody() argument
165 MakeBroadcastHlo(induction_var, /*broadcast_dimensions=*/{}, in GatherLoopBody()
|