| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | cumsum_impl.cu | 29 __global__ void LeftMoveSum(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size_… in LeftMoveSum()
51 __global__ void RightMoveSum(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size… in RightMoveSum()
72 __global__ void CumSumKernelReverse(const T *input, T *output, size_t dim0, size_t dim1, size_t dim… in CumSumKernelReverse()
94 __global__ void CumSumKernel(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size… in CumSumKernel()
115 void CumSum(const T *input, T *output, T *workspace, size_t dim0, size_t dim1, size_t dim2, size_t … in CumSum()
|
| D | cumprod_impl.cu | 29 __global__ void LeftMoveProd(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size… in LeftMoveProd()
51 __global__ void RightMoveProd(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, siz… in RightMoveProd()
72 __global__ void CumProdKernelReverse(const T *input, T *output, size_t dim0, size_t dim1, size_t di… in CumProdKernelReverse()
94 __global__ void CumProdKernel(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, siz… in CumProdKernel()
115 void CumProd(const T *input, T *output, T *workspace, size_t dim0, size_t dim1, size_t dim2, size_t… in CumProd()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/control/ |
| D | tensorlist_fromtensor.cc | 47 int dim0 = tensor_shape[i + 1]; in IsCompatibleShape() local
81 int dim0 = input0_->shape()[0]; in Run() local
|
| D | tensorlist_stack.cc | 45 int dim0 = output0_->shape()[0]; in CheckParam() local
113 size_t dim0 = shape.size(); in MergeSubShape() local
|
| D | tensorlist_getitem.cc | 44 int dim0 = input0->ElementsNum(); in Run() local
|
| D | tensorlist_setitem.cc | 78 int dim0 = output0_->ElementsNum() - 1; in Run() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | common_func.c | 19 int Offset(const int *shape, const int dim0, const int dim1, const int dim2, const int dim3) { in Offset()
23 int OffsetComm(const int *shape, const int dim0, const int dim1, const int dim2) { in OffsetComm()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | cumsum_cpu_kernel.cc | 113 void CumSumCPUKernel::LeftMove(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, si… in LeftMove()
132 void CumSumCPUKernel::RightMove(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, s… in RightMove()
151 void CumSumCPUKernel::Copy(T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size_t strid… in Copy()
165 void CumSumCPUKernel::CumSumKernelReverse(const T *input, T *output, size_t dim0, size_t dim1, size… in CumSumKernelReverse()
184 void CumSumCPUKernel::CumSumKernel(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2… in CumSumKernel()
|
| D | cpu_kernel.cc | 67 size_t CPUKernelUtils::CalcOffset(const std::vector<size_t> &shape, size_t dim0, size_t dim1, size_… in CalcOffset()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/control/ |
| D | tensorlist_fromtensor_infer.c | 40 int dim0 = input0->shape_[0]; in TensorListFromTensorInferShape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | strided_slice_fp32.c | 82 int32_t dim0, dim1, dim2, dim3, dim4, dim5, dim6, dim7; in DoStridedSliceIntFp64Bool() local
144 int32_t dim0, dim1, dim2, dim3, dim4, dim5, dim6, dim7; in DoStridedSlice() local
|
| /third_party/glslang/glslang/HLSL/ |
| D | hlslParseables.cpp | 78 bool IsIllegalSample(const glslang::TString& name, const char* argOrder, int dim0) in IsIllegalSample()
213 … AppendTypeName(glslang::TString& s, const char* argOrder, const char* argType, int dim0, int dim1) in AppendTypeName()
334 …ar /* retOrder */, char /* retType */, char argOrder, char /* argType */, int dim0, int /* dim1 */) in IsValid() argument
909 … for (int dim0 = dim0Min; dim0 <= dim0Max; ++dim0) { // for each dim 0... in initialize() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | tensor_list_from_tensor.cc | 35 int64_t dim0 = input0_shape[0]; in TensorListFromTensorInferShape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | slice_base.c | 119 for (int32_t dim0 = param->begin_[0]; dim0 < param->end_[0]; ++dim0) { in DoSliceNoParallel() local
|