Searched defs:DepthwiseImplementation (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/ |
| H A D | depthwise_implementation.hpp | 38 struct DepthwiseImplementation struct
40 const DepthwiseMethod method;
41 const char *name;
42 std::function<bool(const DepthwiseArgs &, const OutputStage &)> is_supported;
43 std::function<uint64_t(const DepthwiseArgs &, const OutputStage &)> cycle_estimate;
44 …epthwiseCommon<TInput, TWeight, TOutput> *(const DepthwiseArgs &, const OutputStage &)> initialise;
46 bool get_is_supported(const DepthwiseArgs &args, const OutputStage &os) const in get_is_supported()
51 uint64_t get_cycle_estimate(const DepthwiseArgs &args, const OutputStage &os) const in get_cycle_estimate()
56 …mon<TInput, TWeight, TOutput> *get_instance(const DepthwiseArgs &args, const OutputStage &os) const in get_instance()
|