Home
last modified time | relevance | path

Searched defs:conv_type (Results 1 – 10 of 10) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/tools/optimizer/fusion/
Dconv_bn_fusion_test.cc72 MetaGraphTptr BuildCaffeGraph(schema::PrimitiveType conv_type) { in BuildCaffeGraph()
153 MetaGraphTptr BuildTFGraph(schema::PrimitiveType conv_type) { in BuildTFGraph()
Dconv_scale_fusion_test.cc83 MetaGraphTptr BuildGraph(schema::PrimitiveType conv_type, bool conv_with_bias) { in BuildGraph()
Dconv_biasadd_fusion_test.cc71 MetaGraphTptr BuildGraph(schema::PrimitiveType conv_type, schema::PrimitiveType add_type) { in BuildGraph()
Dconv_activation_fusion_test.cc71 MetaGraphTptr BuildGraph(schema::PrimitiveType conv_type, schema::ActivationType activation_type) { in BuildGraph()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/convolution/
Dconv2d_cudnn_gpu_kernel.h38 …explicit ConvolutionCudnnGpuKernel(enum ConvType conv_type) : AbstractConvolutionGpuKernel(conv_ty… in ConvolutionCudnnGpuKernel()
530 auto conv_type = get_conv_type(); in InitSizeLists() local
548 auto conv_type = get_conv_type(); in Set4DDesc() local
Dabstract_conv_gpu_kernel.h37 explicit AbstractConvolutionGpuKernel(enum ConvType conv_type) : conv_type_(conv_type) {} in AbstractConvolutionGpuKernel()
Ddepth_wise_gpu_kernel.h34 …explicit ConvolutionDepthWiseGpuKernel(enum ConvType conv_type) : AbstractConvolutionGpuKernel(con… in ConvolutionDepthWiseGpuKernel()
/third_party/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_format_aos.c540 struct lp_type fetch_type, conv_type; in lp_build_fetch_rgba_aos() local
625 struct lp_type conv_type; in lp_build_fetch_rgba_aos() local
Dlp_bld_format_soa.c165 struct lp_type conv_type = lp_uint_type(type); in lp_build_extract_soa_chan() local
615 struct lp_type conv_type = type; in lp_build_fetch_rgba_soa() local
/third_party/mesa3d/src/gallium/drivers/zink/nir_to_spirv/
Dnir_to_spirv.c2261 SpvId conv_type = get_alu_type(ctx, type, 1, nir_src_bit_size(alu->src[0].src)); in emit_alu() local
2289 SpvId conv_type = get_alu_type(ctx, type, 1, bit_size); in emit_alu() local
2310 SpvId conv_type = get_alu_type(ctx, type, 1, bit_size); in emit_alu() local