Searched refs:aixs_index (Results 1 – 1 of 1) sorted by relevance
590 constexpr size_t aixs_index = 2; in InferImplGatherV2() local591 if (args_spec_list[aixs_index]->isa<AbstractTensor>()) { in InferImplGatherV2()592 auto axis = args_spec_list[aixs_index]->cast<AbstractTensorPtr>(); in InferImplGatherV2()599 } else if (args_spec_list[aixs_index]->isa<AbstractScalar>()) { in InferImplGatherV2()600 auto axis = args_spec_list[aixs_index]->cast<AbstractScalarPtr>(); in InferImplGatherV2()