Searched refs:trt_nb_dims (Results 1 – 1 of 1) sorted by relevance
529 Status ConvertAxis(int tf_axis, int trt_nb_dims, absl::string_view node_name, in ConvertAxis() argument531 const int tf_nb_dims = trt_nb_dims + (use_implicit_batch ? 1 : 0); in ConvertAxis()