| /third_party/mindspore/mindspore/numpy/ |
| D | utils_const.py | 81 def _check_start_normalize(start, ndim): argument
91 def _check_axes_range(axes, ndim): argument
149 def _check_axis_in_range(axis, ndim): argument
159 def _check_axis_valid(axes, ndim): argument
183 def _tile_size(shape, out_shape, ndim): argument
311 def _canonicalize_axis(axis, ndim): argument
356 def _expanded_shape(ndim, axis_size, axis): argument
365 def _add_unit_axes(shape, ndim, append=False): argument
|
| D | array_ops.py | 734 def _atleast_xd(ndim, arys): argument
1097 def _get_moved_perm(ndim, source, destination): argument
1450 def convert_neg_indices(indices, ndim): argument
1614 def _get_flip_start(ndim, shape, axes): argument
1619 def _get_flip_end(ndim, shape, axes): argument
1624 def _get_flip_strides(ndim, axes): argument
|
| D | utils.py | 78 def _expand(x, ndim, axis=0): argument
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | fullconnection_tests.cc | 26 … std::vector<int> *bias_shape, std::vector<int> *output_shape, int ndim, int ci, int co, in CreateParameter()
55 int ndim = 2; variable
74 int ndim = 4; variable
96 int ndim = 3; variable
119 int ndim = 3; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | broadcast_to_infer.c | 67 const int *input_shape1, int *ndim, int *in_shape0, int *in_shape1) { in MakeUpInputShapes()
100 int BroadCastOutputShape(const int *in_shape0, const int *in_shape1, const int ndim, int *out_shape, in BroadCastOutputShape()
120 const int *input_shape1, int *ndim, int *out_shape, bool *has_broad_cast) { in BroadCastToShape()
159 int ndim = input_shape_size; in BroadcastToInferShape() local
|
| D | arithmetic_infer.c | 39 … const int *input_shape1, int *ndim, int *in_shape0, int *in_shape1, int *out_shape, in BroadCastInferShape()
80 int ndim = input_shape0_size; in ArithmeticInferShape() local
|
| D | bias_grad_infer.c | 36 size_t ndim = inshape_size; in BiasGradInferShape() local
|
| D | strided_slice_grad_infer.c | 35 void ApplyBeginEndEllipsisMask(size_t ndim, int *begins, const uint32_t *const begins_mask, int *en… in ApplyBeginEndEllipsisMask()
|
| D | strided_slice_infer.c | 81 int GenerateAxes(const TensorC *axes_tensor, int *axes, int num, int ndim) { in GenerateAxes()
109 int HandleAxesInputExist(const TensorC *const *inputs, int *ndim, int *in_shape, int *begins, int *… in HandleAxesInputExist()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | nnacl_common.h | 26 static inline void ComputeStrides(const int *shape, int *strides, const int ndim) { in ComputeStrides()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | pad.cc | 104 int ndim = in_tensors_.front()->shape().size(); in SetConstArgs() local
|
| /third_party/mindspore/mindspore/ |
| D | _checkparam.py | 651 def check_transpose_axis(axes, ndim): argument
700 def check_swapaxes_axis(axes, ndim): argument
754 def check_axis_in_range(axis, ndim): argument
763 def check_axis_valid(axes, ndim): argument
790 def expanded_shape(ndim, axis_size, axis): argument
841 def check_and_canonicalize_axes(axes, ndim): argument
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | timedistributed.py | 38 def _check_expand_dims_axis(time_axis, ndim, prim_name=None): argument
|
| /third_party/python/Lib/test/ |
| D | test_buffer.py | 264 def strides_from_shape(ndim, shape, itemsize, layout): argument
321 def getindex(ndim, ind, strides): argument
435 def verify_structure(memlen, itemsize, ndim, shape, strides, offset): argument
564 def randslice_from_shape(ndim, shape): argument
698 def genslices_ndim(ndim, shape): argument
716 def rslices_ndim(ndim, shape, iterations=5): argument
774 ndim, shape, strides, argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | arithmetic_fp16.cc | 149 void ArithmeticFP16CPUKernel::TileConstTensor(const void *in_data, void *out_data, size_t ndim, con… in TileConstTensor()
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | math_ops.py | 708 def _tile_size(shape, out_shape, ndim): argument
723 def _expand(x, ndim): argument
|
| /third_party/python/Objects/ |
| D | memoryobject.c | 362 copy_rec(const Py_ssize_t *shape, Py_ssize_t ndim, Py_ssize_t itemsize, in copy_rec()
630 memory_alloc(int ndim) in memory_alloc()
701 int ndim) in mbuf_add_incomplete_view()
1259 copy_shape(Py_ssize_t *shape, const PyObject *seq, Py_ssize_t ndim, in copy_shape()
1298 cast_to_ND(PyMemoryViewObject *mv, const PyObject *shape, int ndim) in cast_to_ND()
1373 Py_ssize_t ndim = 1; in memoryview_cast_impl() local
2086 tolist_rec(const char *ptr, Py_ssize_t ndim, const Py_ssize_t *shape, in tolist_rec()
2793 Py_ssize_t ndim, const Py_ssize_t *shape, in cmp_rec()
|
| /third_party/boost/boost/mpi/ |
| D | cartesian_communicator.hpp | 132 cartesian_topology(int ndim) in cartesian_topology()
|
| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/fp32/ |
| D | arithmetic_fp32_wrapper.c | 17 void TileConstTensor(const float *in_data, float *out_data, size_t ndim, const int *in_shape, const… in TileConstTensor()
|
| /third_party/boost/libs/mpi/test/ |
| D | cartesian_topology_test.cpp | 169 int const ndim = world.size() >= 24 ? 3 : 2; in BOOST_AUTO_TEST_CASE() local
|
| /third_party/mindspore/mindspore/ccsrc/pybind_api/ir/ |
| D | tensor_py.cc | 162 ssize_t ndim() const override { return buffer_.ndim; } in ndim() function in mindspore::tensor::TensorDataNumpy
261 const auto ndim = shape.size(); in GetStrides() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/ |
| D | utils.cc | 187 int GetBroadcastGpuAxis(int ndim, int ori_axis) { in GetBroadcastGpuAxis()
|
| D | opencl_kernel.cc | 226 size_t ndim = global_size_.size(); in GenerateTuningParam() local
|
| /third_party/python/Modules/ |
| D | _testbuffer.c | 473 copy_rec(const Py_ssize_t *shape, Py_ssize_t ndim, Py_ssize_t itemsize, in copy_rec()
612 const Py_ssize_t *suboffsets, Py_ssize_t ndim, Py_ssize_t itemsize) in unpack_rec()
931 Py_ssize_t ndim) in verify_structure()
1121 Py_ssize_t ndim) in init_structure()
1181 Py_ssize_t ndim; in init_ndbuf() local
2483 Py_ssize_t ndim) in arraycmp()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | arithmetic_int8.c | 23 void TileOneDimensionInt8(const int8_t *inData, int8_t *outData, int dim, size_t ndim, const int *i… in TileOneDimensionInt8()
|