Searched refs:z_shape (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ops/operations/ |
| D | comm_ops.py | 272 def infer_shape(self, x_shape, z_shape): argument
278 def infer_dtype(self, x_dtype, z_shape): argument
303 def infer_shape(self, x_shape, z_shape): argument
762 def infer_shape(self, x_shape, z_shape): argument
765 def infer_dtype(self, x_dtype, z_shape): argument
889 def infer_shape(self, x_shape, z_shape): argument
892 def infer_dtype(self, x_dtype, z_shape): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | arithmetic_simplify.cc | 127 ShapeVector x_shape, z_shape; in operator ()() local
147 z_shape = z_abstract->shape()->shape(); in operator ()()
154 z_shape = z_tensor->shape(); in operator ()()
157 if (x_shape != z_shape) { in operator ()()
|