| /third_party/mindspore/mindspore-src/source/tests/st/dyn_shape_dev/ |
| D | test_ops_gather.py | 24 def gather_forward_func(input_params, input_indices, axis, batch_dims=0): argument
29 def gather_backward_func(input_params, input_indices, axis, batch_dims=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | fused_sparse_adam.cc | 26 void ComputeAdam(CpuKernelContext &ctx, MultiThreadComputeParams *input_params, size_t start, size_… in ComputeAdam()
54 void ComputeMomentum(CpuKernelContext &, MultiThreadComputeParams *input_params, size_t start, size… in ComputeMomentum()
65 void ComputeWeight(CpuKernelContext &, MultiThreadComputeParams *input_params, size_t start, size_t… in ComputeWeight()
123 MultiThreadComputeParams input_params; in Compute() local
|
| D | fused_sparse_ftrl.cc | 26 void ComputeFtrl(CpuKernelContext &ctx, MultiThreadComputeParams *input_params, size_t start, size_… in ComputeFtrl()
116 MultiThreadComputeParams input_params; in Compute() local
|
| D | fused_sparse_lazy_adam.cc | 25 void ComputeLazyAdam(CpuKernelContext &ctx, MultiThreadComputeParams *input_params, size_t start, s… in ComputeLazyAdam()
107 MultiThreadComputeParams input_params; in Compute() local
|
| D | fused_sparse_proximal_adagrad.cc | 37 void ComputeProximalAdagrad(CpuKernelContext &ctx, MultiThreadComputeParams *input_params, size_t s… in ComputeProximalAdagrad()
100 MultiThreadComputeParams input_params; in DoCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | sparse_apply_adam_cpu_kernel.cc | 47 void ComputeAdam(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeAdam()
78 void ComputeMomentum(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeMomentum()
91 void ComputeWeight(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeWeight()
292 MultiThreadComputeParams<T> input_params; in LaunchKernel() local
|
| D | sparse_apply_ftrl_cpu_kernel.cc | 48 void ComputeFtrl(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeFtrl()
268 MultiThreadComputeParams<T> input_params; in LaunchKernel() local
448 MultiThreadComputeParams<S> input_params; in LaunchKernel() local
|
| D | sparse_apply_adagrad_cpu_kernel.cc | 41 void ComputeAdaGrad(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeAdaGrad()
227 MultiThreadComputeParams<T> input_params; in LaunchKernel() local
|
| D | sparse_apply_proximal_adagrad_cpu_kernel.cc | 45 void ComputeProximalAdagrad(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeProximalAdagrad()
256 MultiThreadComputeParams<T> input_params; in LaunchKernel() local
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 48 void ComputeLazyAdam(MultiThreadComputeParams<T> *input_params, size_t start, size_t end) { in ComputeLazyAdam()
265 MultiThreadComputeParams<T> input_params; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/error_manager/ |
| D | aicore_error.py | 13 def construct(self, input_params, input_indices, axis): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_embedding_lookup_op.py | 30 def construct(self, input_params, input_indices, offset): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/ |
| D | relux_int8.cc | 42 const auto &input_params = input->quant_params(); in Prepare() local
|
| D | l2_norm_int8.cc | 54 const auto &input_params = input->quant_params(); in Prepare() local
|
| D | layer_norm_int8.cc | 54 const auto &input_params = input->quant_params(); in SetQuantArgs() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_gather.cc | 37 …auto input_params = std::make_shared<abstract::AbstractTensor>(param.dtype, param.input_params_sha… in TEST_P() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_asd.py | 53 def __call__(self, input_params, input_indices, axis): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/ |
| D | acl_convert.cc | 329 const std::vector<TensorParams> &input_params, in ConvertValueDependToHostInput()
488 const std::vector<TensorParams> &input_params) { in ConvertToAclInput()
630 … const std::string &kernel_name, std::vector<TensorParams> *input_params) { in ConvertAttrToAclInput()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/ |
| D | acl_kernel_mod.cc | 47 … SetParamsDataTypeIfComplexInput(const PrimitivePtr &prim, std::vector<TensorParams> *input_params, in SetParamsDataTypeIfComplexInput()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/ |
| D | top_cell.h | 50 OrderedMap<std::string, ParameterPtr> input_params; // Hold input parameters member
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ir/ |
| D | func_graph_cloner.cc | 585 … AnfNodeWeakPtrList *const lift_params, AnfNodeWeakPtrList *const input_params) { in AddParameters()
714 AnfNodeWeakPtrList input_params; in LiftParameters() local
|
| D | func_graph.cc | 88 AnfNodePtrList input_params; in get_inputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/parse/ |
| D | resolve.cc | 205 std::vector<AnfNodePtr> input_params; in ConvertLoadedGraph() local
|