| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | conv3d_gpu_kernel.h | 301 const int kDims = 5; in SetNDDesc() local
302 int dimA[kDims]; in SetNDDesc()
303 int strideAin[kDims]; in SetNDDesc()
304 int dimAout[kDims]; in SetNDDesc()
305 int strideAout[kDims]; in SetNDDesc()
306 int filterDimA[kDims]; in SetNDDesc()
307 SetDimA(in_shape, dimA, kDims, data_format_); in SetNDDesc()
308 SetStrideA(in_shape, strideAin, kDims, data_format_); in SetNDDesc()
309 SetDimA(output_shape, dimAout, kDims, data_format_); in SetNDDesc()
310 SetStrideA(output_shape, strideAout, kDims, data_format_); in SetNDDesc()
[all …]
|
| D | conv3d_grad_input_gpu_kernel.h | 322 const int kDims = 5; in SetNDDesc() local
323 int dimA[kDims]; in SetNDDesc()
324 int strideAin[kDims]; in SetNDDesc()
325 int dimAdy[kDims]; in SetNDDesc()
326 int strideAdy[kDims]; in SetNDDesc()
327 int filterDimA[kDims]; in SetNDDesc()
328 SetDimA(input_shape, dimA, kDims, data_format_); in SetNDDesc()
329 SetStrideA(input_shape, strideAin, kDims, data_format_); in SetNDDesc()
330 SetDimA(dy_shape, dimAdy, kDims, data_format_); in SetNDDesc()
331 SetStrideA(dy_shape, strideAdy, kDims, data_format_); in SetNDDesc()
[all …]
|
| D | conv3d_grad_filter_gpu_kernel.h | 348 const int kDims = 5; in SetNDDesc() local
349 int dimA[kDims]; in SetNDDesc()
350 int strideAin[kDims]; in SetNDDesc()
351 int dimAdy[kDims]; in SetNDDesc()
352 int strideAdy[kDims]; in SetNDDesc()
353 int filterDimA[kDims]; in SetNDDesc()
354 SetDimA(in_shape, dimA, kDims, data_format_); in SetNDDesc()
355 SetStrideA(in_shape, strideAin, kDims, data_format_); in SetNDDesc()
356 SetDimA(dy_shape, dimAdy, kDims, data_format_); in SetNDDesc()
357 SetStrideA(dy_shape, strideAdy, kDims, data_format_); in SetNDDesc()
[all …]
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | tile_fusion.cc | 25 void TileFusion::set_dims(const std::vector<int64_t> &dims) { (void)this->AddAttr(kDims, MakeValue(… in set_dims()
28 auto value_ptr = GetAttr(kDims); in get_dims()
|
| /third_party/skia/tests/ |
| D | ProxyConversionTest.cpp | 130 static constexpr SkISize kDims = {64, 64}; in DEF_GPUTEST_FOR_RENDERING_CONTEXTS() local
136 format, kDims, GrRenderable::kYes, 1, GrMipmapped::kNo, SkBackingFit::kApprox, in DEF_GPUTEST_FOR_RENDERING_CONTEXTS()
150 format, kDims, GrRenderable::kYes, 1, GrMipmapped::kNo, SkBackingFit::kApprox, in DEF_GPUTEST_FOR_RENDERING_CONTEXTS()
164 format, kDims, GrRenderable::kNo, 1, GrMipmapped::kNo, SkBackingFit::kApprox, in DEF_GPUTEST_FOR_RENDERING_CONTEXTS()
|
| D | ProgramsTest.cpp | 250 static constexpr SkISize kDims = {34, 18}; in ProgramUnitTest() local
253 auto proxy = proxyProvider->createProxy(format, kDims, GrRenderable::kYes, 1, in ProgramUnitTest()
261 static constexpr SkISize kDims = {16, 22}; in ProgramUnitTest() local
264 auto proxy = proxyProvider->createProxy(format, kDims, GrRenderable::kNo, 1, mipMapped, in ProgramUnitTest()
|
| D | GrSurfaceTest.cpp | 108 static constexpr SkISize kDims = {64, 64}; in DEF_GPUTEST_FOR_ALL_CONTEXTS() local
131 sk_sp<GrSurface> tex = createTexture(kDims, combo.fColorType, combo.fFormat, in DEF_GPUTEST_FOR_ALL_CONTEXTS()
143 combo.fFormat, kDims, GrRenderable::kNo, 1, GrMipmapped::kYes, in DEF_GPUTEST_FOR_ALL_CONTEXTS()
156 kDims, combo.fFormat, GrTextureType::k2D, GrRenderable::kYes, 1, in DEF_GPUTEST_FOR_ALL_CONTEXTS()
169 kDims, combo.fFormat, GrTextureType::k2D, GrRenderable::kYes, 2, in DEF_GPUTEST_FOR_ALL_CONTEXTS()
|
| D | TextureBindingsResetTest.cpp | 77 static constexpr SkISize kDims = {10, 10}; in DEF_GPUTEST_FOR_GL_RENDERING_CONTEXTS() local
80 auto tex = gpu->createTexture(kDims, format, GrTextureType::k2D, GrRenderable::kNo, 1, in DEF_GPUTEST_FOR_GL_RENDERING_CONTEXTS()
|
| D | OpChainTest.cpp | 178 static constexpr SkISize kDims = {kNumOps + 1, 1}; variable
185 format, kDims, GrRenderable::kYes, 1, GrMipmapped::kNo, SkBackingFit::kExact,
|
| D | WritePixelsTest.cpp | 531 static constexpr SkISize kDims = {32, 64}; in DEF_GPUTEST_FOR_RENDERING_CONTEXTS() local
537 format, kDims, GrRenderable::kNo, 1, GrMipmapped::kNo, SkBackingFit::kApprox, in DEF_GPUTEST_FOR_RENDERING_CONTEXTS()
|
| D | ProcessorTest.cpp | 147 static constexpr SkISize kDims = {10, 10}; in DEF_GPUTEST_FOR_ALL_CONTEXTS() local
161 format, kDims, GrRenderable::kNo, 1, GrMipmapped::kNo, SkBackingFit::kExact, in DEF_GPUTEST_FOR_ALL_CONTEXTS()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | op_utils.h | 221 constexpr auto kDims = "dims"; variable
|