Searched refs:kCpuFP32SubGraph (Results 1 – 4 of 4) sorted by relevance
68 subgraph_type_ = kCpuFP32SubGraph; in SubGraphKernel()127 subgraph_type_ = kCpuFP32SubGraph; in CpuSubGraph()143 subgraph_type_ = kCpuFP32SubGraph; in CpuFp32SubGraph()
126 …if (belong_subgraph_type != kernel::kCpuFP32SubGraph && belong_subgraph_type != kernel::kCpuFP16Su… in CastKernelWeight()146 …f (tensor->data_type() == kNumberTypeFloat16 && belong_subgraph_type == kernel::kCpuFP32SubGraph) { in CastKernelWeight()1170 if (type == kernel::kCpuFP32SubGraph) { in CreateSubGraphKernel()1226 return kernel::kCpuFP32SubGraph; in GetKernelSubGraphType()1231 return kernel::kCpuFP32SubGraph; in GetKernelSubGraphType()1330 kernel::SubGraphType cur_sub_graph_type = kernel::kCpuFP32SubGraph; in SchedulePartialToSubGraphKernel()1502 case kernel::SubGraphType::kCpuFP32SubGraph: { in KernelFitCurrentSubGraph()1643 return kernel::kCpuFP32SubGraph; in PartialSubGraphType()
70 kCpuFP32SubGraph, enumerator
38 if (kernel->subgraph_type() == mindspore::kernel::kCpuFP32SubGraph) { in CheckRuntime()