Searched defs:axis_input (Results 1 – 2 of 2) sorted by relevance
37 auto axis_input = cnode->input(kNameGatherInputNum - 1); in Mapper() local
211 auto axis_input = GetValue<int64_t>(ops[iter_ops]->input_value().at(2)); in PrepareGatherV2() local254 auto axis_input = GetValue<int64_t>(ops[iter_ops]->input_value().at(2)); in PrepareGatherV2P() local