Searched defs:input_param (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_lamb_apply_optimizer_assign.py | 32 …def construct(self, grad, input_param, beta_1, one_minus_beta_1, beta_2, one_minus_beta_2, epsilon, argument
38 def get_output(grad, inputv, inputm, input_param, beta_1, one_minus_beta_1, beta_2, one_minus_beta_… argument
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/ |
| D | acl_model_multi.cc | 165 auto input_param = std::dynamic_pointer_cast<Parameter>(in); in SetInputs() local
|
| /third_party/mindspore/mindspore/core/load_mindir/ |
| D | load_model.cc | 167 std::fstream input_param(path, std::ios::in | std::ios::binary); in ParseGraphProto() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/format/ |
| D | to_format_base.cc | 212 auto input_param = input->cast<ParameterPtr>(); in HandleGraphInput() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | cpu_session.cc | 188 auto input_param = input_node->cast<ParameterPtr>(); in LoadInputData() local
|
| D | gpu_session.cc | 306 auto input_param = input_node->cast<ParameterPtr>(); in UpdateGraphInputAbstract() local
|
| D | ascend_session.cc | 348 auto input_param = parameter->cast<ParameterPtr>(); in TensorNeedSync() local
475 auto input_param = input_node->cast<ParameterPtr>(); in LoadInputData() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 564 auto input_param = node->cast<ParameterPtr>(); in AssignStaticMemoryInput() local
1586 auto input_param = AnfAlgo::GetPrevNodeOutput(kernel, 0, true); in GetFirstPSEmbeddingCache() local
1660 auto input_param = AnfAlgo::GetPrevNodeOutput(kernel, 0, true); in CheckIfSupportPSEmbeddingCache() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_kernel_runtime.cc | 347 auto input_param = item->cast<ParameterPtr>(); in BindInputTensorAddressPtr() local
|
| /third_party/mindspore/mindspore/ |
| D | _checkparam.py | 866 def check_input_format(input_param): argument
|
| /third_party/mindspore/third_party/proto/caffe/ |
| D | caffe.proto | 387 optional InputParameter input_param = 143; field
|