Searched defs:input_params (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | sparse_apply_adam_cpu_kernel.cc | 28 void ComputeAdam(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeAdam()
58 void ComputeMomentum(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeMomentum()
71 void ComputeWeight(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeWeight()
179 MultiThreadComputeParams<T> input_params; in LaunchKernel() local
|
| D | sparse_apply_ftrl_cpu_kernel.cc | 28 void ComputeFtrl(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeFtrl()
163 MultiThreadComputeParams<T> input_params; in LaunchKernel() local
|
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 28 void ComputeProximalAdagrad(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeProximalAdagrad()
152 MultiThreadComputeParams<T> input_params; in LaunchKernel() local
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 28 void ComputeLazyAdam(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeLazyAdam()
156 MultiThreadComputeParams<T> input_params; in LaunchKernel() local
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_embedding_lookup_op.py | 30 def construct(self, input_params, input_indices, offset): argument
|
| /third_party/mindspore/tests/ut/cpp/optimizer/ad/ |
| D | kpynative_test.cc | 84 auto input_params = primal_fg->parameters(); in BuildBpropFuncGraph() local
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | func_graph_cloner.cc | 302 AnfNodePtrList *const lift_params, AnfNodePtrList *const input_params) { in AddParameters()
485 AnfNodePtrList input_params; in LiftParameters() local
|
| D | func_graph.cc | 74 std::vector<AnfNodePtr> input_params; in get_inputs() local
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/ |
| D | resolve.cc | 162 std::vector<AnfNodePtr> input_params; in ConvertLoadedGraph() local
|