Searched defs:src_dim (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_vmap/ |
| D | vmap_convolution_ops.py | 158 def _get_merge_shape(src_dim, dst_dim, shape): argument
165 def _reshape_merge_dims(src_dim, dst_dim, target): argument
175 def _get_expand_shape(src_dim, dst_size, shape, prim_name): argument
183 def _reshape_expand_dims(src_dim, dst_size, target, prim_name): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/mkldnn/ |
| D | mkl_cpu_kernel.cc | 64 size_t src_dim = src_shape.size(); in GetPadding() local
98 size_t src_dim = src_shape.size(); in GetPadding() local
|
| D | conv_cpu_kernel.cc | 54 auto src_dim = src_shape.size(); in Resize() local
|
| D | conv3d_transpose_cpu_kernel.cc | 59 size_t src_dim = src_shape.size(); in Resize() local
|
| D | conv_grad_filter_cpu_kernel.cc | 70 size_t src_dim = src_shape.size(); in Resize() local
|
| D | conv_grad_input_cpu_kernel.cc | 71 size_t src_dim = src_shape.size(); in Resize() local
|
| D | pooling_cpu_kernel_nnacl.cc | 56 void PoolingCpuKernelNnaclMod::GetPadList(size_t src_dim, size_t padlist_len) { in GetPadList()
155 auto src_dim = in_size_.size(); in Resize() local
|
| D | pooling_cpu_kernel.cc | 98 const size_t src_dim = src_shape.size(); in Resize() local
|
| D | pooling_grad_cpu_kernel.cc | 112 const size_t src_dim = src_shape.size(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | maxpool_grad_grad_cpu_kernel.cc | 83 const size_t src_dim = in_shapes_.size(); in CheckInputVaild() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/ |
| D | trt_op_converter.cc | 48 const nvinfer1::Dims &src_dim = input->tensor()->getDimensions(); in ToShape() local
|