Searched refs:subgraph_type (Results 1 – 8 of 8) sorted by relevance
35 if (kernel->subgraph_type() == mindspore::kernel::kGpuFp32SubGraph) { in CheckRuntime()38 if (kernel->subgraph_type() == mindspore::kernel::kCpuFP32SubGraph) { in CheckRuntime()95 if (kernel->subgraph_type() != mindspore::kernel::kCpuFP16SubGraph) { in CheckRuntime2()
125 MS_ASSERT(kernel->subgraph_type() == kernel::kNotSubGraph); in CastKernelWeight()193 MS_ASSERT(kernel->subgraph_type() == kernel::kNotSubGraph); in HandleBuildinCpuKernelWeight()225 auto subgraph_type = kernel->subgraph_type(); in InitKernels() local226 if (subgraph_type == kernel::kNotSubGraph) { in InitKernels()232 auto ret = HandleBuildinCpuKernelWeight(subgraph_type, node); in InitKernels()357 MS_LOG(DEBUG) << "[subgraph] : " << subgraph->name() << ", type:" << subgraph->subgraph_type(); in Schedule()1477 bool KernelFitCurrentSubGraph(const kernel::SubGraphType subgraph_type, const kernel::LiteKernel &k… in KernelFitCurrentSubGraph() argument1478 switch (subgraph_type) { in KernelFitCurrentSubGraph()1529 if (cur_kernel->subgraph_type() != kernel::kNotSubGraph || in FindAllSubGraphKernels()1560 if (cur_kernel->subgraph_type() != kernel::kNotSubGraph) { in ConstructSubGraphs()[all …]
98 if (kernel->subgraph_type() != kernel::kNotSubGraph) { in ReSize()
480 if (kernel->subgraph_type() == kernel::kNotSubGraph) { in FreePackOpWeight()938 …if (kernel->subgraph_type() == kernel::kGpuFp16SubGraph || kernel->subgraph_type() == kernel::kGpu… in ReSizeKernels()
328 SubGraphType subgraph_type() const { return this->subgraph_type_; } in subgraph_type() function
224 if (is_fp16 && subgraph_type() == kGpuFp16SubGraph) { in UpdateTensorDataTypePass()304 ocl_runtime_->SetFp16Enable(subgraph_type() == kGpuFp16SubGraph); in Prepare()385 if (kernel->subgraph_type() != kernel::kNotSubGraph) { in ReSize()
167 if (kernel->subgraph_type() != kernel::kNotSubGraph) { in Nc4hw4PassAct()
618 if (ori_kernel->subgraph_type() == kernel::kNotSubGraph) { in CompileTrainKernels()