Searched refs:ignore_first_dim (Results 1 – 2 of 2) sorted by relevance
142 bool ignore_first_dim) { in TensorShapeToTrtDims() argument144 const int offset = (ignore_first_dim ? 1 : 0); in TensorShapeToTrtDims()
248 bool ignore_first_dim = false) { in TensorShapeArrayToTrtDims() argument251 *out = TensorShapeToTrtDims(tensor_shape, ignore_first_dim); in TensorShapeArrayToTrtDims()