Home
last modified time | relevance | path

Searched defs:CpuKernelContext (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dfused_sparse_adam.cc54 void ComputeMomentum(CpuKernelContext &, MultiThreadComputeParams *input_params, size_t start, size… in ComputeMomentum() argument
65 void ComputeWeight(CpuKernelContext &, MultiThreadComputeParams *input_params, size_t start, size_t… in ComputeWeight() argument
Dsparse_cross.cc97 std::string SparseTensorColumn<std::string>::Feature(CpuKernelContext &, int64_t batch, int64_t n) … in Feature() argument