Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!641 pynative-lamb-op-zeros-like-tensor-query-failed 

Merge pull request !641 from JoyLvliang/pynative-lamb-op-zeros-like-tensor-query-failed
This commit is contained in:
mindspore-ci-bot 2020-04-26 14:28:55 +08:00 committed by Gitee
parent 7c7d95acf9 52ab6f4948
commit c1ef1e0aee
4 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/_extends/builtin_operations.py
View File

@ -86,7 +86,7 @@ def identity(x):
def zeros_like_tensor(x):
"""Implement `zeros_like_tensor`."""
x = x.asnumpy()
value = Tensor(np.zeros(x.shape))
value = Tensor(np.zeros(x.shape).astype(np.float32))
return value

1
mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.h
View File

@ -26,7 +26,6 @@ void AscendDataLayout(const std::shared_ptr<session::KernelGraph> &kernel_graph)
void AscendMixPrecision(const std::shared_ptr<session::KernelGraph> &kernel_graph);
void AscendBackendOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph);
void AscendBackendIRFusionOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph);
void RunOpAscendBackendIRFusionOptimization(const std::shared_ptr<session::KernelGraph> &kernel_graph);
} // namespace opt
} // namespace mindspore

2
mindspore/ccsrc/pynative/pynative_execute.cc
View File

@ -39,7 +39,7 @@
const char SINGLE_OP_GRAPH[] = "single_op_graph";
// primitive unable to infer value for constant input in PyNative mode
const std::unordered_set<std::string> vm_operators = {"partial", "depend", "make_ref"};
const std::set<std::string> vm_operators = {"partial", "depend", "make_ref", "zeros_like_tensor"};
namespace mindspore {
namespace pynative {

9
mindspore/ops/operations/debug_ops.py
View File

@ -45,6 +45,9 @@ class ScalarSummary(Primitive):
def __init__(self):
"""init"""
def __call__(self, *args, **kwargs):
pass
class ImageSummary(Primitive):
"""
@ -70,6 +73,9 @@ class ImageSummary(Primitive):
def __init__(self):
"""init"""
def __call__(self, *args, **kwargs):
pass
class TensorSummary(Primitive):
"""
@ -97,6 +103,9 @@ class TensorSummary(Primitive):
def __init__(self):
"""init"""
def __call__(self, *args, **kwargs):
pass
class HistogramSummary(Primitive):
"""