Searched refs:ignore_first_dim (Results 1 – 1 of 1) sorted by relevance
203 bool ignore_first_dim) { in TensorShapeToTrtDims() argument205 const int offset = (ignore_first_dim ? 1 : 0); in TensorShapeToTrtDims()215 bool ignore_first_dim = false) { in TensorShapeArrayToTrtDims() argument218 *out = TensorShapeToTrtDims(tensor_shape, ignore_first_dim); in TensorShapeArrayToTrtDims()