31CircleNode *CircleBCQGatherGraphBuilder::build_node(
const circle::OperatorT &op,
32 const std::vector<CircleNode *> &inputs,
38 node->input_binary(inputs.at(1));
39 node->indices(inputs.at(2));
40 node->input_clusters(inputs.at(3));
42 const auto *options = op.builtin_options.AsBCQGatherOptions();
43 node->input_hidden_size(options->input_hidden_size);
44 node->axis(options->axis);
bool validate(const ValidateArgs &args) const final
loco::Node * input_scales(void) const
bool validate(const ValidateArgs &args, size_t input_cnt) const