Searched refs:ConvertAxis (Results 1 – 1 of 1) sorted by relevance
504 Status ConvertAxis(int tf_axis, int trt_nb_dims, absl::string_view node_name, in ConvertAxis() function3726 TF_RETURN_IF_ERROR(ConvertAxis(axis[0], inputs.at(0).GetTrtDims().nbDims, in ConvertGather()