Home
last modified time | relevance | path

Searched defs:dim1 (Results 1 – 16 of 16) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dcumsum_impl.cu29 __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()
Dcumprod_impl.cu29 __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/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dcommon_func.c19 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/boost/libs/units/test/
Dfail_base_dimension.cpp27 struct dim1 : boost::units::base_dimension<dim1, 1> {}; struct
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcumsum_cpu_kernel.cc113 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()
Dcpu_kernel.cc67 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/fp32/
Dstrided_slice_fp32.c82 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/mindspore/mindspore/lite/src/runtime/kernel/arm/control/
Dtensorlist_fromtensor.cc48 int dim1 = elements_shape[i]; in IsCompatibleShape() local
Dtensorlist_stack.cc114 size_t dim1 = output_shape_.size(); in MergeSubShape() local
/third_party/glslang/glslang/HLSL/
DhlslParseables.cpp213 … AppendTypeName(glslang::TString& s, const char* argOrder, const char* argType, int dim0, int dim1) in AppendTypeName()
910 … for (int dim1 = dim1Min; dim1 <= dim1Max; ++dim1) { // for each dim 1... in initialize() local
/third_party/boost/libs/geometry/test/algorithms/detail/sections/
Dsectionalize.cpp207 typedef boost::mpl::vector_c<std::size_t, 0> dim1; in test_sectionalize() typedef
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dslice_base.c131 for (int dim1 = param->begin_[1]; dim1 < param->end_[1]; ++dim1) { in DoSliceNoParallel() local
/third_party/boost/boost/geometry/srs/
Dprojection.hpp78 static const std::size_t dim1 = geometry::dimension<Point1>::value; in copy_higher_dimensions() local
/third_party/boost/boost/units/detail/
Ddimension_impl.hpp196 typedef typename Begin1::item dim1; typedef
/third_party/skia/third_party/externals/angle2/src/tests/gl_tests/
DGLSLTest.cpp10729 for (int dim1 = 0; dim1 < 2; ++dim1) in TEST_P() local
10838 for (int dim1 = 0; dim1 < 2; ++dim1) in TEST_P() local
10971 for (int dim1 = 0; dim1 < 2; ++dim1) in TEST_P() local
11148 for (int dim1 = 0; dim1 < 2; ++dim1) in TEST_P() local
11278 for (int dim1 = 0; dim1 < 2; ++dim1) in TEST_P() local
/third_party/libxml2/
Dxmlregexp.c441 xmlRegCalloc2(size_t dim1, size_t dim2, size_t elemSize) { in xmlRegCalloc2()