Searched defs:loop_state_shape (Results 1 – 3 of 3) sorted by relevance
343 const Shape& loop_state_shape) { in BuildConditionComputation()405 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in BuildDependentBodyComputation2() local478 const Shape& loop_state_shape = in BuildIndependentBodyComputation() local665 const Shape& loop_state_shape, HloInstruction* data_init, in BuildWhileInstructionWithCustomInit()1294 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in TEST_F() local1355 const Shape loop_state_shape = in TEST_F() local1629 const Shape loop_state_shape = in TEST_F() local1692 const Shape loop_state_shape = in TEST_F() local1749 const Shape loop_state_shape = ShapeUtil::MakeTupleShape( in TEST_F() local1928 const Shape loop_state_shape = in MakeBenchmarkWhileBody() local[all …]
158 MakeCountedLoopConditionComputation(const Shape& loop_state_shape, in MakeCountedLoopConditionComputation()182 const Shape& loop_state_shape, in MakeCountedLoopBodyComputation()255 Shape loop_state_shape = MakeLoopStateShapeWithLayout(init_values); in MakeCountedLoop() local
1258 Shape loop_state_shape = ShapeUtil::MakeTupleShape( in BM_WhileLoop() local