Searched refs:default_tensor (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/graph_utils/ |
| D | graph_pattern.py | 181 def __init__(self, para_name, default_tensor, requires_grad=False, layerwise_parallel=False): argument
193 self.default_tensor = default_tensor
196 …if isinstance(para_name, str) and isinstance(default_tensor, Tensor) and isinstance(requires_grad,…
198 NewParameter_.__init__(self, self.para_name, self.default_tensor, self.requires_grad,
202 layerwise_parallel(bool)， got : {para_name}, {default_tensor}, \
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | py_pass.cc | 105 auto default_value = new_para_pattern->default_tensor(); in BuildNewParameter()
210 …auto default_tensor = std::make_shared<tensor::Tensor>(default_input->data_type(), default_input->… in ReflectParamBackToPython() local
214 …py::object new_parameter = parameter_class(default_tensor, param_name, requires_grad, layerwise_pa… in ReflectParamBackToPython()
277 auto default_value = new_para_pattern->default_tensor(); in Run()
|
| D | pattern.h | 218 …explicit NewParameter(string para_name, tensor::TensorPtr default_tensor, bool requires_grad, bool… in NewParameter() argument
221 default_tensor_ = std::make_shared<tensor::Tensor>(*default_tensor.get()); in NewParameter()
230 tensor::TensorPtr default_tensor() { return default_tensor_; } in default_tensor() function
|
| /third_party/mindspore/tests/ut/python/optimizer/ |
| D | test_python_pass.py | 312 default_tensor = Tensor(np.ones((4, 4)), mindspore.float32)
313 new_para = NewParameter("Merlin", default_tensor)
|