Home
last modified time | relevance | path

Searched defs:input_param (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/pi_jit/control_flow/
Dtest_control_while.py21 def construct(self, x, y, input_param): argument
94 def construct(self, x, y, input_param): argument
169 def construct(self, x, y, input_param): argument
231 def construct(self, x, y, input_param): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_sparse_embedding_operation.cc156 …ParameterPtr input_param = NewMapParameter(graph, kNumberTypeInt32, kNumberTypeFloat32, kOneDimens… in BuildEmbeddingCacheLookupKernel() local
185 …ParameterPtr input_param = NewMapParameter(graph, kNumberTypeInt32, kNumberTypeFloat32, kOneDimens… in BuildEmbeddingCacheUpdateKernel() local
213 …ParameterPtr input_param = NewMapParameter(graph, kNumberTypeInt32, kNumberTypeFloat32, kOneDimens… in BuildEmbeddingCacheEraseKernel() local
Ddevice_dense_embedding_operation.cc225 ParameterPtr input_param = NewParameter(graph, kFloat32, kTwoDimensionalShape); in BuildEmbeddingCacheLookupKernel() local
259 ParameterPtr input_param = NewParameter(graph, kFloat32, kTwoDimensionalShape); in BuildEmbeddingCacheUpdateKernel() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/
Dtest_lamb_apply_optimizer_assign.py32 …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-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dps_embedding_cache_inserter.cc286 ParameterPtr input_param = graph->add_parameter(); in ConstructEmbeddingLookupSubGraph() local
336 ParameterPtr input_param = graph->add_parameter(); in ConstructUpdateEmbeddingSubGraph() local
381 const ParameterPtr &input_param, in CreateEmbeddingLookupKernel()
412 const ParameterPtr &input_param, in CreateEmbeddingUpdateKernel()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dembedding_cache_scheduler.cc80 auto input_param = common::AnfAlgo::GetPrevNodeOutput(kernel, 0, true); in GetFirstEmbeddingCacheTableInfo() local
174 auto input_param = common::AnfAlgo::GetPrevNodeOutput(kernel, 0, true); in CheckGraphValidForEmbeddingCache() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/
Dlamb_apply_weight_assign.cc42 const auto &input_param = inputs[4]; in Expand() local
Dlamb_apply_optimizer_assign.cc32 const auto &input_param = inputs[3]; in Expand() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/model/acl/
Dacl_model_multi.cc172 auto input_param = std::dynamic_pointer_cast<Parameter>(in); in SetInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/parser/
Dunify_format.cc141 auto input_param = input->cast<api::ParameterPtr>(); in HandleGraphInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/
Dcpu_session.cc191 auto input_param = input_node->cast<ParameterPtr>(); in LoadInputData() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/format/
Dto_format_base.cc231 auto input_param = input->cast<ParameterPtr>(); in HandleGraphInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/
Dgpu_session.cc299 auto input_param = input_node->cast<ParameterPtr>(); in UpdateGraphInputAbstract() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dcpu_kernel_runtime.cc351 auto input_param = item->cast<ParameterPtr>(); in BindInputTensorAddressPtr() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dop_runner.cc91 auto input_param = input_node->cast<ParameterPtr>(); in UpdateParameterShapeFromInputTensor() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/
Dacl_convert.cc539 auto input_param = GetInputParam(inputs, input_on_host_, ms_real_idx); in ConvertToAclInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Ddata_prepare_actor.cc421 auto input_param = input_node->cast<ParameterPtr>(); in UpdateDynamicShapeAndSize() local
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/
D_checkparam.py1246 def check_input_format(input_param): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Ddevice_address_utils.cc257 auto input_param = item->cast<ParameterPtr>(); in CreateParameterDeviceAddress() local
Dkernel_runtime.cc661 auto input_param = item->cast<ParameterPtr>(); in GetDeviceAddress() local
/third_party/mindspore/mindspore-src/source/third_party/proto/caffe/
Dcaffe.proto387 optional InputParameter input_param = 143; field
/third_party/mindspore/mindspore-src/source/mindspore/core/load_mindir/
Dload_model.cc2589 std::fstream input_param(path, std::ios::in | std::ios::binary); in ParseGraphProto() local