Home
last modified time | relevance | path

Searched refs:get_shape (Results 1 – 25 of 27) sorted by relevance

12

/third_party/mindspore/tests/ut/cpp/dataset/
Drandom_crop_with_bbox_op_test.cc52 TensorShape get_shape = output_tensor_row_[0]->shape(); in TEST_F() local
53 actual = get_shape[0] * get_shape[1] * get_shape[2]; in TEST_F()
88 TensorShape get_shape = output_tensor_row_[0]->shape(); in TEST_F() local
89 actual = get_shape[0] * get_shape[1] * get_shape[2]; in TEST_F()
/third_party/boost/boost/python/numpy/
Dufunc.hpp57 Py_intptr_t const * get_shape() const;
111 : zeros(in_array.get_nd(), in_array.get_shape(), out_dtype); in call()
176 : zeros(iter.get_nd(), iter.get_shape(), out_dtype); in call()
Dndarray.hpp113 Py_intptr_t const * get_shape() const { return get_struct()->shape; } in get_shape() function in boost::python::numpy::ndarray
/third_party/mindspore/tests/st/ops/ascend/
Dtest_conv2dGradFilter.py34 self.get_shape = P.Shape()
38 return self.conv2d_grad(out_, x_, self.get_shape(self.y))
/third_party/mindspore/tests/st/ops/gpu/
Dtest_conv2d_backprop_input_op.py42 self.get_shape = P.Shape()
46 return self.conv_input(out, w, self.get_shape(x))
Dtest_conv2d_backprop_filter_op.py43 self.get_shape = P.Shape()
47 return self.conv_filter(out, x, self.get_shape(w))
/third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/
Dtest_conv2d_backprop_input.py56 self.get_shape = P.Shape()
60 return self.conv_input(self.out, self.w, self.get_shape(self.x))
Dtest_conv2d_transpose.py56 self.get_shape = P.Shape()
60 return self.conv_input(self.out, self.w, self.get_shape(self.x))
Dtest_conv2d_backprop_filter.py57 self.get_shape = P.Shape()
61 return self.conv_filter(self.out, self.x, self.get_shape(self.w))
/third_party/mindspore/tests/st/ops/cpu/
Dtest_conv2d_backprop_input_op.py57 self.get_shape = P.Shape()
60 return self.conv_input(self.out, self.w, self.get_shape(self.x))
Dtest_conv2d_backprop_filter_op.py58 self.get_shape = P.Shape()
61 return self.conv_filter(self.out, self.x, self.get_shape(self.w))
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_nn_ops.py57 get_shape = P.Shape()
61 x_shape = get_shape(x)
62 w_shape = get_shape(w)
88 get_shape = P.Shape()
91 dx = input_grad(w, dout, get_shape(x))
92 dw = filter_grad(x, dout, get_shape(w))
123 get_shape = P.Shape()
142 x_shape = get_shape(x)
150 out_shape = get_shape(out)
186 get_shape = P.Shape()
[all …]
/third_party/mindspore/tests/ut/python/pipeline/infer/
Dtest_auto_monad.py215 shape1 = self.get_shape(x)
221 def get_shape(self, x): member in test_unused_var.UnusedVar
242 shape1 = self.hof_get_shape(self.get_shape, x)
251 def get_shape(self, x): member in test_hof_unused_var.UnusedVar
279 return F.partial(self.get_shape, x)
281 def get_shape(self, x): member in test_partial_hof_unused_var.UnusedVar
/third_party/mindspore/tests/ut/python/parallel/
Dtest_dropout_do_mask.py30 self.get_shape = P.Shape()
37 shape = self.get_shape(out)
Dtest_matmul_dropout.py72 self.get_shape = P.Shape()
90 shape = self.get_shape(x)
/third_party/boost/libs/python/doc/numpy/reference/
Dmulti_iter.rst30 Py_intptr_t const * get_shape() const;
84 Py_intptr_t const * get_shape() const;
Dndarray.rst50 Py_intptr_t const * get_shape() const;
314 Py_intptr_t const * get_shape() const;
/third_party/boost/libs/python/doc/html/numpy/_sources/reference/
Dmulti_iter.rst.txt30 Py_intptr_t const * get_shape() const;
84 Py_intptr_t const * get_shape() const;
/third_party/mindspore/mindspore/core/ops/
Dbroadcast_to.h46 std::vector<int64_t> get_shape() const;
Dbroadcast_to.cc83 std::vector<int64_t> BroadcastTo::get_shape() const { in get_shape() function in mindspore::ops::BroadcastTo
/third_party/boost/libs/python/src/numpy/
Dufunc.cpp55 Py_intptr_t const * multi_iter::get_shape() const in get_shape() function in boost::python::numpy::multi_iter
/third_party/mindspore/mindspore/offline_debug/
Ddbg_services.py284 … tensor_data = TensorData(b'', elem.get_data_size(), elem.get_dtype(), elem.get_shape())
286 …or_data = TensorData(elem.get_data_ptr(), elem.get_data_size(), elem.get_dtype(), elem.get_shape())
613 return self.instance.get_shape()
/third_party/mindspore/mindspore/nn/layer/
Dconv.py413 get_shape = P.Shape()
416 weight_init_shape = get_shape(weight_init)
1208 get_shape = P.Shape()
1211 weight_init_shape = get_shape(weight_init)
/third_party/mindspore/mindspore/ccsrc/debug/debugger/offline_debug/
Dmi_pybind_register.cc67 .def("get_shape", &tensor_data_t::get_shape); in PYBIND11_MODULE()
Ddbg_services.h115 const std::vector<int64_t> &get_shape() const { return shape; } in get_shape() function

12