Home
last modified time | relevance | path

Searched defs:wei (Results 1 – 11 of 11) sorted by relevance

/external/ComputeLibrary/src/runtime/heuristics/direct_conv/
DClDirectConvDefaultConfigBifrost.cpp45 …irectConvDefaultConfigBifrost::configure(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure()
72 …vDefaultConfigBifrost::configure_G71_f32(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G71_f32()
96 …vDefaultConfigBifrost::configure_G71_f16(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G71_f16()
120 …nvDefaultConfigBifrost::configure_G71_u8(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G71_u8()
144 …aultConfigBifrost::configure_default_f32(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_default_f32()
168 …aultConfigBifrost::configure_default_f16(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_default_f16()
DClDirectConvDefaultConfigValhall.cpp45 …irectConvDefaultConfigValhall::configure(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure()
73 …vDefaultConfigValhall::configure_G78_f32(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G78_f32()
134 …vDefaultConfigValhall::configure_G78_f16(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G78_f16()
246 …nvDefaultConfigValhall::configure_G78_u8(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G78_u8()
270 …vDefaultConfigValhall::configure_G57_f32(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G57_f32()
330 …vDefaultConfigValhall::configure_G57_f16(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G57_f16()
/external/ComputeLibrary/src/runtime/heuristics/dwc_native/
DClDWCNativeDefaultConfigBifrost.cpp38 DWCComputeKernelInfo configure_f32(const ITensorInfo *src, const ITensorInfo *wei, const PadStrideI… in configure_f32()
108 DWCComputeKernelInfo configure_f16(const ITensorInfo *src, const ITensorInfo *wei, const PadStrideI… in configure_f16()
201 …DWCNativeDefaultConfigBifrost::configure(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure()
230 …eDefaultConfigBifrost::configure_G71_f32(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G71_f32()
236 …eDefaultConfigBifrost::configure_G71_f16(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G71_f16()
242 …eDefaultConfigBifrost::configure_G7x_f32(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G7x_f32()
248 …eDefaultConfigBifrost::configure_G7x_f16(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G7x_f16()
254 …veDefaultConfigBifrost::configure_G7x_u8(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G7x_u8()
DClDWCNativeDefaultConfigValhall.cpp41 …DWCNativeDefaultConfigValhall::configure(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure()
71 …eDefaultConfigValhall::configure_G78_f32(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G78_f32()
133 …eDefaultConfigValhall::configure_G78_f16(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G78_f16()
212 …veDefaultConfigValhall::configure_G78_u8(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G78_u8()
237 …eDefaultConfigValhall::configure_G77_f16(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G77_f16()
/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/
DGpuDepthwiseConv2d.cpp45 …_and_init_dst_if_empty(ITensorInfo *dst, const ITensorInfo *src, const ITensorInfo *wei, const Dep… in calculate_and_init_dst_if_empty()
69 const ITensorInfo *wei, in is_supported_op_helper()
143 const ITensorInfo *wei, in is_supported_op()
152 const ITensorInfo *wei, in validate_op()
188 ITensorInfo *wei, in create_op()
DGpuConv2d.cpp101 …_and_init_dst_if_empty(ITensorInfo *dst, const ITensorInfo *src, const ITensorInfo *wei, const Con… in calculate_and_init_dst_if_empty()
119 const ITensorInfo *wei, in is_supported_op_helper()
181 const ITensorInfo *wei, in is_supported_op()
190 const ITensorInfo *wei, in validate_op()
229 ITensorInfo *wei, in create_op()
/external/ComputeLibrary/src/runtime/heuristics/indirect_conv/
DClIndirectConvDefaultConfigValhall.cpp43 …irectConvDefaultConfigValhall::configure(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure()
60 …vDefaultConfigValhall::configure_G77_f32(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G77_f32()
103 …vDefaultConfigValhall::configure_G77_f16(const ITensorInfo *src, const ITensorInfo *wei, const Pad… in configure_G77_f16()
/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/components/cl/
DClComponentDirectConv2d.cpp79 const auto wei = tensors.get_const_tensor(TensorType::ACL_SRC_1); in validate() local
DClComponentDepthwiseConv2d.cpp114 const auto wei = tensors.get_const_tensor(TensorType::ACL_SRC_1); in validate() local
/external/ComputeLibrary/src/cpu/kernels/directconv2d/nchw/
Dall.cpp106 Iterator wei(weights, window_w); in convolve_nchw() local
/external/ComputeLibrary/src/cpu/kernels/directconv2d/nhwc/neon/
Dimpl.cpp95 Iterator wei(weights, window_w); in convolve_nhwc() local