Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dreduce.cc174 const std::vector<Axis> all_axis = {Axis::WIDTH, Axis::HEIGHT, Axis::DEPTH, in GetReduceKernelCode() local
176 for (const auto& a : all_axis) { in GetReduceKernelCode()
273 for (const auto& a : all_axis) { in GetReduceKernelCode()
281 for (const auto& a : all_axis) { in GetReduceKernelCode()
327 for (const auto& a : all_axis) { in GetReduceKernelCode()
408 for (const auto& a : all_axis) { in GetReduceKernelCode()