Searched refs:tf_nb_dims (Results 1 – 1 of 1) sorted by relevance
531 const int tf_nb_dims = trt_nb_dims + (use_implicit_batch ? 1 : 0); in ConvertAxis() local533 if (tf_axis < -tf_nb_dims || tf_axis >= tf_nb_dims) { in ConvertAxis()536 -tf_nb_dims, ", ", tf_nb_dims, "), at ", node_name); in ConvertAxis()539 if (tf_axis < 0) tf_axis += tf_nb_dims; in ConvertAxis()