Searched refs:input_shape_with_layout (Results 1 – 3 of 3) sorted by relevance
444 Shape input_shape_with_layout(ashape); in TEST_F() local446 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({1, 0}); in TEST_F()452 ShapeLayout(input_shape_with_layout); in TEST_F()585 Shape input_shape_with_layout = in TEST_F() local588 HloInstruction::CreateParameter(0, input_shape_with_layout, "param")); in TEST_F()622 Shape input_shape_with_layout(input_shape); in TEST_F() local623 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({2, 1, 0}); in TEST_F()625 HloInstruction::CreateParameter(0, input_shape_with_layout, "param")); in TEST_F()639 Shape input_shape_with_layout(input_shape); in TEST_F() local640 *input_shape_with_layout.mutable_layout() = LayoutUtil::MakeLayout({2, 1, 0}); in TEST_F()[all …]
831 for (const ShapeProto& shape_proto : arg->input_shape_with_layout()) { in Compile()
397 repeated ShapeProto input_shape_with_layout = 3; field