Searched refs:input_shape_with_layout (Results 1 – 3 of 3) sorted by relevance
408 Shape input_shape_with_layout(ashape); in TEST_F() local410 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({1, 0}); in TEST_F()416 ShapeLayout(input_shape_with_layout); in TEST_F()545 Shape input_shape_with_layout = in TEST_F() local548 HloInstruction::CreateParameter(0, input_shape_with_layout, "param")); in TEST_F()582 Shape input_shape_with_layout(input_shape); in TEST_F() local583 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({2, 1, 0}); in TEST_F()585 HloInstruction::CreateParameter(0, input_shape_with_layout, "param")); in TEST_F()599 Shape input_shape_with_layout(input_shape); in TEST_F() local600 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({2, 1, 0}); in TEST_F()[all …]
870 for (const ShapeProto& shape_proto : arg->input_shape_with_layout()) { in Compile()
609 repeated ShapeProto input_shape_with_layout = 3; field