Searched defs:interleaved_parallel (Results 1 – 3 of 3) sorted by relevance
190 def __init__(self, hidden_size, ffn_hidden_size, dp, mp, interleaved_parallel): argument
112 … const Shape &tensor_shape, bool interleaved_parallel, bool check_device_num) { in InitFromExtendVector()
2407 … std::vector<std::vector<int64_t>> *tensor_map_vector, bool *interleaved_parallel) { in GetLayoutFromAttrValue()2469 bool interleaved_parallel; in ExtractUserConfigLayout() local2498 bool interleaved_parallel; in ExtractUserConfigLayout() local