Searched refs:parameter_shape (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_using_seed_for_initializer.py | 21 parameter_shape = [16, 4] variable
27 …self.para_xavier_uniform = Parameter(init.initializer('xavier_uniform', parameter_shape), name="xa…
28 …self.para_he_uniform = Parameter(init.initializer('he_uniform', parameter_shape), name="he_uniform…
29 … self.para_xavier_uniform2 = Parameter(init.initializer(init.XavierUniform(), parameter_shape),
31 …self.para_he_uniform2 = Parameter(init.initializer(init.HeUniform(), parameter_shape), name="he_un…
32 … self.para_truncated_normal = Parameter(init.initializer(init.TruncatedNormal(), parameter_shape),
34 … self.para_normal = Parameter(init.initializer(init.Normal(), parameter_shape), name="normal")
35 … self.para_uniform = Parameter(init.initializer(init.Uniform(), parameter_shape), name="uniform")
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | ascend_inference_session.cc | 143 auto parameter_shape = AnfAlgo::GetOutputDeviceShape(parameter, 0); in CompareInput() local
153 …!is_scalar_shape(trans_input) || !is_scalar_shape(parameter_shape)) && (trans_input != parameter_s… in CompareInput()
155 << ", but the parameter shape is " << PrintInputShape(parameter_shape) in CompareInput()
|
| D | gpu_inference_session.cc | 143 auto parameter_shape = AnfAlgo::GetOutputDeviceShape(parameter, 0); in CompareInput() local
153 …!is_scalar_shape(trans_input) || !is_scalar_shape(parameter_shape)) && (trans_input != parameter_s… in CompareInput()
155 << ", but the parameter shape is " << PrintInputShape(parameter_shape) in CompareInput()
|
| D | session_basic.cc | 1772 auto parameter_shape = AnfAlgo::GetOutputDeviceShape(parameter, 0); in GetModelInputsInfo() local
1773 …(void)std::transform(parameter_shape.begin(), parameter_shape.end(), std::back_inserter(input_shap… in GetModelInputsInfo()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.cc | 358 auto parameter_shape = GetNodeShape(parameter); in HandleNoUsedParameter() local
359 if (parameter_shape.empty()) { in HandleNoUsedParameter()
362 Shape slice_shape = parameter_shape[0]; in HandleNoUsedParameter()
|
| /third_party/mindspore/mindspore/train/ |
| D | serialization.py | 1270 parameter_shape = sliced_parameters[0].data.shape
1271 parameter_shape_length = len(parameter_shape)
1283 or parameter.data.shape[1:] != parameter_shape[1:]:
1287 if parameter.data.shape != parameter_shape:
|