Searched defs:split_param (Results 1 – 3 of 3) sorted by relevance
32 auto split_param = op_->mutable_splitnd(); in BuildLayer() local
159 auto *split_param = reinterpret_cast<SplitParameter *>(split_param_base); in DoSplitReduceConcatFusion() local213 SplitParameter *split_param, in CreateCustomNode()
23 const SplitParameter *split_param, int data_size) { in DoSplit()