Searched defs:adasum_param_tensor_layout_map (Results 1 – 2 of 2) sorted by relevance
1272 … std::unordered_map<std::string, std::shared_ptr<TensorLayout>> *adasum_param_tensor_layout_map) { in HandleAdaSum()1366 std::unordered_map<std::string, std::shared_ptr<TensorLayout>> *adasum_param_tensor_layout_map) { in HandleMirrorInAdaSum()
3664 auto adasum_param_tensor_layout_map = AdaSumParamTensorLayout(root); in ParallelPartProcess() local