Home
last modified time | relevance | path

Searched defs:PostProcess (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dinfershape_pass.h38 …virtual STATUS PostProcess(const FuncGraphPtr &func_graph, const CNodePtr &cnode) { return lite::R… in PostProcess() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/cuda/
Dunique.cc31 int UniqueCudaKernel::PostProcess() { in PostProcess() function in mindspore::kernel::UniqueCudaKernel
Dcuda_kernel.cc65 int CudaKernel::PostProcess() { in PostProcess() function in mindspore::kernel::CudaKernel
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/auto_grad/
Dbackward_function.h60 virtual ValuePtrList PostProcess(const ValuePtrList &gradient_value) { return gradient_value; } in PostProcess() function
Dnative_backward_function.cc61 ValuePtrList NativeBackwardFunc::PostProcess(const ValuePtrList &gradient_value) { in PostProcess() function in mindspore::pijit::grad::NativeBackwardFunc
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Dslice_tensorrt.h34 virtual nvinfer1::ITensor *PostProcess(TensorRTContext *ctx, nvinfer1::ITensor *input, in PostProcess() function
Dslice_tensorrt.cc138 nvinfer1::ITensor *PostProcess(TensorRTContext *ctx, nvinfer1::ITensor *input, in PostProcess() function in mindspore::lite::__anon761179f60111::StrideSliceTensorRTUtil
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/const_fold/
Dfold_along_infershape.cc26 STATUS ConstFoldAlongInferShape::PostProcess(const FuncGraphPtr &func_graph, const CNodePtr &cnode)… in PostProcess() function in mindspore::opt::ConstFoldAlongInferShape
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/kernel/
Didentity_kernel.cc78 int IdentityKernel::PostProcess() { return lite::RET_OK; } in PostProcess() function in mindspore::kernel::IdentityKernel
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/ascend_native/
Dascend_native_copy_kernel.cc200 int AscendNativeCopyKernel::PostProcess() { in PostProcess() function in mindspore::kernel::AscendNativeCopyKernel
Dascend_native_composite_kernel.cc449 int AscendNativeCompositeKernel::PostProcess() { in PostProcess() function in mindspore::kernel::AscendNativeCompositeKernel
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/
Dvariable.cc44 ValuePtrList BackwardNode::PostProcess(const ValuePtrList &gradient_value) { in PostProcess() function in mindspore::pynative::autograd::BackwardNode
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/
Dbase_kernel.h120 virtual int PostProcess() { in PostProcess() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/ascend_native/
Dascend_native_base_kernel.h74 int PostProcess() override { return mindspore::lite::RET_OK; } in PostProcess() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/distributed/cluster/
Dcluster_context.cc250 void ClusterContext::PostProcess() { in PostProcess() function in mindspore::distributed::cluster::ClusterContext
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dgraph_kernel_optimization.cc163 GkPassManagerPtr GraphKernelOptimizer::PostProcess() const { in PostProcess() function in mindspore::graphkernel::GraphKernelOptimizer
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/
Dgraph_kernel_optimization.cc295 PassManagerPtr GraphKernelOptimizer::PostProcess() const { in PostProcess() function in mindspore::graphkernel::GraphKernelOptimizer
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dlite_kernel.h73 virtual int PostProcess() { return FreeInWorkTensor(); } in PostProcess() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/
Dopencl_kernel.h288 int PostProcess() override { in PostProcess() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dlinear_sum_assignment.cc341 void LinearSumAssignmentCpuKernel::PostProcess(int64_t *a, int64_t *b, bool transpose, uint64_t ele… in PostProcess() function in aicpu::LinearSumAssignmentCpuKernel
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dmul_reduce_fusion.cc135 int MulReduceFusion::PostProcess(const FuncGraphPtr &func_graph) { in PostProcess() function in mindspore::opt::MulReduceFusion
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dlinear_sum_assignment_cpu_kernel.cc364 void LinearSumAssignmentCpuKernelMod::PostProcess(int64_t *a, int64_t *b, const std::vector<int64_t… in PostProcess() function in mindspore::kernel::LinearSumAssignmentCpuKernelMod
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/executor/
Dsub_graph_kernel.cc561 int CpuFp16SubGraph::PostProcess() { in PostProcess() function in mindspore::kernel::CpuFp16SubGraph
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/
Dbprop.cc425 void BpropExpander::PostProcess(const CNodePtr &cnode) const { in PostProcess() function in mindspore::expander::bprop::BpropExpander