Searched refs:sens_tensor_param (Results 1 – 1 of 1) sorted by relevance
2362 auto sens_tensor_param = sens_tensor_node->cast<ParameterPtr>(); in SplitSens() local2364 … sens_tensor_param->set_user_data<TensorLayout>(std::make_shared<TensorLayout>(loss_grad_layout)); in SplitSens()2388 auto sens_tensor_param = sens_tensor_node->cast<ParameterPtr>(); in SplitSens() local2389 … sens_tensor_param->set_user_data<TensorLayout>(std::make_shared<TensorLayout>(loss_grad_layout)); in SplitSens()