Searched defs:param_layout (Results 1 – 2 of 2) sorted by relevance
167 TensorLayout input_layout, param_layout, output_layout; in TEST_F() local
1546 TensorLayout param_layout; in GenerateTensorLayoutForParamReshapeWithStra() local1621 auto param_layout = GenerateTensorLayoutForParamReshapeWithStra(node, input_strategy); in FindParameterNextLayout() local