Searched defs:weight_format (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/core/utils/ |
| H A D | AssemblyUtils.cpp | 70 arm_gemm::WeightFormat map_to_arm_gemm_weight_format(const arm_compute::WeightFormat &weight_format) in map_to_arm_gemm_weight_format()
190 …compute::WeightFormat map_to_arm_compute_weight_format(const arm_gemm::WeightFormat &weight_format) in map_to_arm_compute_weight_format()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | ConvolutionLayerFixture.h | 424 …sorInfo prepare_weights(const TensorInfo tensor_info, const arm_compute::WeightFormat weight_forma… in prepare_weights()
458 …rrange_data(const AccessorType src, AccessorType dst, const arm_compute::WeightFormat weight_forma… in rearrange_data()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
| H A D | CpuGemmConv2d.cpp | 102 …ol enable_fast_math, int gemm_3d_depth, bool fixed_format, arm_compute::WeightFormat weight_format) in configure_mm()
182 …h, int gemm_3d_depth, bool skip_im2col, bool fixed_format, arm_compute::WeightFormat weight_format) in validate_mm()
|
| H A D | CpuFullyConnected.cpp | 112 …TensorInfo *dst, const ActivationLayerInfo &act, bool enable_fast_math, WeightFormat weight_format) in validate_mm()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/ |
| H A D | Types.h | 2125 arm_compute::WeightFormat weight_format() const in weight_format() function
2129 void set_weight_format(arm_compute::WeightFormat weight_format) in set_weight_format()
2585 arm_compute::WeightFormat weight_format() const in weight_format() function
2594 void set_weight_format(arm_compute::WeightFormat weight_format) in set_weight_format()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_matmul_op_fused.cc | 122 memory::format_tag weight_format = in Compute() local
|
| H A D | mkl_matmul_ops_common.h | 65 memory::format_tag weight_format; member
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_convolution.py | 3413 weight_format, argument
3729 weight_format, argument
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/assembly/ |
| H A D | arm_gemm.hpp | 112 WeightFormat weight_format = WeightFormat::ANY; member
|
| /aosp_15_r20/external/pytorch/test/xpu/ |
| H A D | test_conv.py | 1107 weight_format, argument
|