27 const auto *node = loco::must_cast<const luci::CircleMean *>(circle_node);
28 assert(node->arity() == 2);
34 auto temp_index_unique =
37 temp_index_unique->set_data_buffer(
nullptr);
41 auto resolved_axes_unique =
44 resolved_axes_unique->set_data_buffer(
nullptr);
48 auto temp_sum_unique =
51 temp_sum_unique->set_data_buffer(
nullptr);
57 return std::make_unique<kernels::Mean>(input, axes, output, temp_index, resolved_axes, temp_sum,