Searched refs:server_kernel (Results 1 – 2 of 2) sorted by relevance
256 bool ParameterAggregator::AssignMemory(const K server_kernel, const CNodePtr &cnode, in AssignMemory() argument259 MS_EXCEPTION_IF_NULL(server_kernel); in AssignMemory()263 const std::vector<std::string> &input_names = server_kernel->input_names(); in AssignMemory()264 const std::vector<size_t> &input_size_list = server_kernel->GetInputSizeList(); in AssignMemory()266 MS_LOG(EXCEPTION) << "Server kernel " << typeid(server_kernel.get()).name() in AssignMemory()387 …late bool ParameterAggregator::AssignMemory(std::shared_ptr<kernel::OptimizerKernel> server_kernel,392 …te bool ParameterAggregator::AssignMemory(std::shared_ptr<kernel::AggregationKernel> server_kernel,
114 bool AssignMemory(const K server_kernel, const CNodePtr &cnode,