Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEBatchNormalizationLayerKernel.cpp49 struct BatchNormalizationSelectorData struct
54 using BatchNormalizationSelectorPtr = std::add_pointer<bool(const BatchNormalizationSelectorData &d…
70 …[](const BatchNormalizationSelectorData & data) { return data.dt == DataType::F16 && data.ci.has_s… in __anon0df002e40202()
75 …[](const BatchNormalizationSelectorData & data) { return data.dt == DataType::F32 && data.ci.has_s… in __anon0df002e40302()
83 [](const BatchNormalizationSelectorData & data) { return data.dt == DataType::F16; }, in __anon0df002e40402()
89 [](const BatchNormalizationSelectorData & data) { return data.dt == DataType::F32; }, in __anon0df002e40502()
95 const BatchNormalizationKernel *get_implementation(const BatchNormalizationSelectorData &data) in get_implementation()
113 …const auto *uk = get_implementation(BatchNormalizationSelectorData{ input->data_type(), CPUInfo::g… in validate_arguments()
391 …const auto *uk = get_implementation(BatchNormalizationSelectorData{ _input->info()->data_type(), C… in run()