Searched defs:loop_state_shape (Results 1 – 3 of 3) sorted by relevance
154 MakeCountedLoopConditionComputation(const Shape& loop_state_shape, in MakeCountedLoopConditionComputation()177 const Shape& loop_state_shape, in MakeCountedLoopBodyComputation()249 Shape loop_state_shape = MakeLoopStateShapeWithLayout(init_values); in MakeCountedLoop() local
414 const Shape& loop_state_shape) { in BuildConditionComputation()476 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in BuildDependentBodyComputation2() local549 const Shape& loop_state_shape = in BuildIndependentBodyComputation() local760 const Shape& loop_state_shape, HloInstruction* data_init, in BuildWhileInstructionWithCustomInit()1448 const Shape& loop_state_shape = ShapeUtil::MakeTupleShape( in TEST_F() local1509 const Shape loop_state_shape = in TEST_F() local1783 const Shape loop_state_shape = in TEST_F() local1846 const Shape loop_state_shape = in TEST_F() local1903 const Shape loop_state_shape = ShapeUtil::MakeTupleShape( in TEST_F() local2082 const Shape loop_state_shape = in MakeBenchmarkWhileBody() local[all …]
1272 Shape loop_state_shape = ShapeUtil::MakeTupleShape( in BM_WhileLoop() local