28 const auto *node = loco::must_cast<const luci::CircleBatchMatMul *>(circle_node);
29 assert(node->arity() == 2);
38 lhs_scratchpad->set_data_buffer(
nullptr);
41 rhs_scratchpad->set_observable(
false);
42 rhs_scratchpad->set_data_buffer(
nullptr);
51 if (execution_plan.offsets().size() > 1)
53 assert(execution_plan.offsets().size() == 3);
56 lhs_scratchpad->set_offset(execution_plan.offsets().at(1));
57 rhs_scratchpad->set_offset(execution_plan.offsets().at(2));
64 params.
adj_x = node->adj_x();
65 params.
adj_y = node->adj_y();
67 return std::make_unique<kernels::BatchMatMul>(lhs, rhs, output, lhs_tmp, rhs_tmp, params);