Browse Source

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

Merge pull request !641 from JoyLvliang/pynative-lamb-op-zeros-like-tensor-query-failed
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
c1ef1e0aee
4 changed files with 11 additions and 3 deletions
  1. +1
    -1
      mindspore/_extends/builtin_operations.py
  2. +0
    -1
      mindspore/ccsrc/pre_activate/ascend/ascend_backend_optimization.h
  3. +1
    -1
      mindspore/ccsrc/pynative/pynative_execute.cc
  4. +9
    -0
      mindspore/ops/operations/debug_ops.py

+ 1
- 1
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




+ 0
- 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



+ 1
- 1
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
- 0
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):
"""


Loading…
Cancel
Save