Searched refs:first_trt_dim (Results 1 – 1 of 1) sorted by relevance
336 const int first_trt_dim = use_implicit_batch ? 1 : 0; in ValidateTensorProperties() local337 for (int d = first_trt_dim; d < shape.dims(); ++d) { in ValidateTensorProperties()349 for (int d = first_trt_dim; d < shape.dims(); ++d) { in ValidateTensorProperties()