Home
last modified time | relevance | path

Searched refs:weight_transform (Results 1 – 13 of 13) sorted by relevance

/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/
Dwinograd_implementations.hpp78 namespace weight_transform { namespace
154 inline std::vector<const weight_transform::ITransform *> get_weight_transforms( in get_weight_transforms()
162 std::vector<const weight_transform::ITransform *> weight_transforms; in get_weight_transforms()
163 for (auto impl = weight_transform::implementation_list<TWeight, TWinogradIn>(); in get_weight_transforms()
263 for (auto weight_transform = weight_transforms.cbegin(); in get_implementation() local
264 !success && weight_transform != weight_transforms.cend(); in get_implementation()
265 weight_transform++) in get_implementation()
269 … if ((*output_transform)->get_input_rows() == (*weight_transform)->get_transformed_tile_rows() && in get_implementation()
270 (*output_transform)->get_input_cols() == (*weight_transform)->get_transformed_tile_cols()) in get_implementation()
283 dest.weight_transform = *weight_transform; in get_implementation()
Dweight_transforms_fp16.cpp32 namespace weight_transform { namespace
Dweight_transforms_fp32.cpp30 namespace weight_transform { namespace
Dweight_transform.hpp33 namespace weight_transform { namespace
/external/ComputeLibrary/src/core/NEON/kernels/assembly/
Dwinograd.hpp95 namespace weight_transform namespace
212 const weight_transform::ITransform *weight_transform = nullptr; member
/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/weight_transforms/
Dcpp_fp32_1x6_1x3.cpp29 namespace weight_transform { namespace
Dcpp_fp32_1x4_1x5.cpp29 namespace weight_transform { namespace
Dcpp_fp32_1x2_1x7.cpp29 namespace weight_transform { namespace
Darm_fp32_2x2_3x3.cpp30 namespace weight_transform { namespace
Darm_fp32_4x4_3x3.cpp30 namespace weight_transform { namespace
Da64_fp16_4x4_3x3.cpp31 namespace weight_transform { namespace
Darm_fp32_2x2_5x5.cpp30 namespace weight_transform { namespace
/external/ComputeLibrary/src/cpu/operators/
DCpuWinogradConv2d.cpp406 _winograd_impl.weight_transform->execute( in prepare()