Searched refs:expected_weight_format (Results 1 – 12 of 12) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEGEMMConvolutionLayer.cpp | 83 Status NEGEMMConvolutionLayer::has_opt_impl(arm_compute::WeightFormat &expected_weight_format, cons… in has_opt_impl() argument
87 …return cpu::CpuGemmConv2d::has_opt_impl(expected_weight_format, src, weights, biases, dst, conv_in… in has_opt_impl()
|
| D | NEFullyConnectedLayer.cpp | 92 Status NEFullyConnectedLayer::has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const… in has_opt_impl() argument
96 …return cpu::CpuFullyConnected::has_opt_impl(expected_weight_format, input, weights, biases, output… in has_opt_impl()
|
| D | NEGEMM.cpp | 87 Status NEGEMM::has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const ITensorInfo *a… in has_opt_impl() argument
91 return cpu::CpuGemm::has_opt_impl(expected_weight_format, a, b, c, output, gemm_info); in has_opt_impl()
|
| /external/ComputeLibrary/src/cpu/operators/internal/ |
| D | CpuGemmAssemblyDispatch.cpp | 652 Status CpuGemmAssemblyDispatch::has_opt_impl(arm_compute::WeightFormat &expected_weight_format, con… in has_opt_impl() argument
663 …ormat arm_gemm_expected_wf = assembly_utils::map_to_arm_gemm_weight_format(expected_weight_format); in has_opt_impl()
717 expected_weight_format = assembly_utils::map_to_arm_compute_weight_format(arm_gemm_expected_wf); in has_opt_impl()
756 arm_compute::WeightFormat expected_weight_format; in validate() local
757 …const Status ret = CpuGemmAssemblyDispatch::has_opt_impl(expected_weight_format, a, b… in validate()
758 if((bool)ret && expected_weight_format != arm_compute::WeightFormat::ANY) in validate()
763 ARM_COMPUTE_RETURN_ERROR_ON_MSG((expected_weight_format != info.weight_format), in validate()
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NEGEMM.h | 95 …static Status has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const ITensorInfo *a…
|
| D | NEGEMMConvolutionLayer.h | 180 …static Status has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const ITensorInfo *s…
|
| D | NEFullyConnectedLayer.h | 151 …static Status has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const ITensorInfo *i…
|
| /external/ComputeLibrary/src/cpu/operators/ |
| D | CpuFullyConnected.h | 105 …static Status has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const ITensorInfo *s…
|
| D | CpuGemmConv2d.h | 126 …static Status has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const ITensorInfo *s…
|
| D | CpuGemm.cpp | 381 Status CpuGemm::has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const ITensorInfo *… in has_opt_impl() argument
386 return CpuGemmAssemblyDispatch::has_opt_impl(expected_weight_format, a, b, c, d, asm_info); in has_opt_impl()
|
| D | CpuFullyConnected.cpp | 345 Status CpuFullyConnected::has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const ITe… in has_opt_impl() argument
354 return CpuGemm::has_opt_impl(expected_weight_format, src, weights, biases, dst, gemm_info); in has_opt_impl()
|
| D | CpuGemmConv2d.cpp | 391 Status CpuGemmConv2d::has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const ITensor… in has_opt_impl() argument
420 return CpuGemm::has_opt_impl(expected_weight_format, src, weights, biases, dst, gemm_info); in has_opt_impl()
|