Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dadaptive_avg_pool3d_impl.cu23 … const uint output_height, const uint output_width, const uint output_depth, in AdaptiveAvgPool3DKernel()
64 … const uint output_height, const uint output_width, const uint output_depth, in ApplyAdaptiveAvgPool3D()
Dadaptive_avg_pool3d_grad_impl.cu34 … const uint output_width, const uint output_depth, T *input_data, in AdaptiveAvgPool3DGradKernel()
74 const uint output_depth, T *input_data, T *output_data, in ApplyAdaptiveAvgPool3DGrad()
Dextract_image_patches_impl.cu29 int64_t output_depth, const T *input, T *output) { in ExtractImagePatches()
69int64_t output_depth, const T *input, T *output, cudaStream_t stream) { in CalExtractImagePatchesNHWC()
Dextract_volume_patches_impl.cu22int64_t output_depth, int64_t output_height, int64_t output_width, bool need_batch, in ExtractVolumePatches()
80int64_t output_depth, int64_t output_height, int64_t output_width, bool need_batch, in CalExtractVolumePatches()
Dadaptive_max_pool3d_impl.cu42 int64_t output_depth = static_cast<int64_t>(output_size[0]); in AdaptiveMaxPool3DKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dupsample_trilinear3d.h43 int64_t output_depth; variable
Dupsample_nearest3d_grad.h40 int64_t output_depth; variable
Dupsample_trilinear3d_grad.h71 int64_t output_depth; variable
Dupsample_nearest3d.cc167 size_t output_depth = static_cast<size_t>(y_shape_[kIndex2]); in UpsampleNearest3dCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dupsample_nearest_3d_cpu_kernel.cc104 int64_t output_depth = y_shape_[kIndex2]; in LaunchKernel() local
Dupsample_nearest_3d_grad_cpu_kernel.cc125 int64_t output_depth = output_shape_[kIndex2]; in LaunchKernel() local
Dupsample_trilinear_3d_grad_cpu_kernel.cc139 int64_t output_depth = output_shape_[kIndex2]; in LaunchKernel() local
Dupsample_trilinear_3d_cpu_kernel.cc124 int64_t output_depth = y_shape_[kIndex2]; in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dmaxpool_grad_grad.c96 const int output_depth = param->output_d_; in MaxPool3DGradGrad() local
/third_party/mesa3d/src/amd/vpelib/src/chip/vpe10/
Dvpe10_mpc.c1174 void vpe10_mpc_set_denorm(struct mpc *mpc, int opp_id, enum color_depth output_depth, in vpe10_mpc_set_denorm()
/third_party/mesa3d/src/gallium/frontends/d3d10umd/
DShaderTGSI.c209 struct ureg_dst output_depth; member