Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/runtime/heuristics/dwc_native/
H A 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()
H A 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()
/aosp_15_r20/external/ComputeLibrary/src/runtime/heuristics/direct_conv/
H A 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()
H A 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()
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/
H A 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()
H A 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()
/aosp_15_r20/external/ComputeLibrary/src/runtime/heuristics/indirect_conv/
H A 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()
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/components/cl/
H A DClComponentDirectConv2d.cpp79 const auto wei = tensors.get_const_tensor(TensorType::ACL_SRC_1); in validate() local
H A DClComponentDepthwiseConv2d.cpp114 const auto wei = tensors.get_const_tensor(TensorType::ACL_SRC_1); in validate() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/directconv2d/nchw/
H A Dall.cpp106 Iterator wei(weights, window_w); in convolve_nchw() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/directconv2d/nhwc/neon/
H A Dimpl.cpp95 Iterator wei(weights, window_w); in convolve_nhwc() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear_prepack.cpp296 ideep::tensor wei = ideep::tensor({w_dims, dnnl::memory::data_type::s8}, weight.data_ptr()); in pack_weight_to_onednn_tensor() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DMatmul.cpp423 auto wei = ideep::tensor( in _mkldnn_matmul_i8i8i32_with_primitive() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DRNN_miopen.cpp242 Tensor permute_wei_for_miopen(Tensor wei, int64_t mode) in permute_wei_for_miopen()
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-1B-Instruct/
H A Dtokenizer.json30616 "wei": 28204, number
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/Llama-3.2-3B-Instruct/
H A Dtokenizer.json30616 "wei": 28204, number
/aosp_15_r20/external/executorch/examples/mediatek/models/llm_models/weights/llama3-8B-instruct/
H A Dtokenizer.json30557 "wei": 28204, number