Searched defs:loop_state_shape (Results 1 – 3 of 3) sorted by relevance
152 MakeCountedLoopConditionComputation(const Shape& loop_state_shape, in MakeCountedLoopConditionComputation()175 const Shape& loop_state_shape, in MakeCountedLoopBodyComputation()234 Shape loop_state_shape = MakeLoopStateShape(init_values); in MakeCountedLoop() local
414 const Shape& loop_state_shape) { in BuildConditionComputation()473 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in BuildDependentBodyComputation2() local543 const Shape& loop_state_shape = in BuildIndependentBodyComputation() local754 const Shape& loop_state_shape, HloInstruction* data_init, in BuildWhileInstructionWithCustomInit()1238 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in TEST_F() local1299 const Shape loop_state_shape = in TEST_F() local1580 const Shape loop_state_shape = in TEST_F() local1643 const Shape loop_state_shape = in TEST_F() local1700 const Shape loop_state_shape = ShapeUtil::MakeTupleShape( in TEST_F() local1879 const Shape loop_state_shape = in MakeBenchmarkWhileBody() local[all …]
1273 Shape loop_state_shape = ShapeUtil::MakeTupleShape( in BM_WhileLoop() local