Searched defs:max_rank (Results 1 – 6 of 6) sorted by relevance
127 int max_rank = std::max(lhs_shape.dims(), rhs_shape.dims()); in Compile() local
204 int max_rank = 1; in CwiseOutputElementCount() local
2382 const int64_t max_rank = 4; in PropagateShapes() local
158 int64_t max_rank = -1; in VerifyOperandsHaveSameShapesOrBroadcastableShape() local
233 const int max_rank = nvinfer1::Dims::MAX_DIMS + (use_implicit_batch ? 1 : 0); in ValidateTensorProperties() local
760 std::vector<int64_t> OpTest::RandomDims(int min_rank, int max_rank, in RandomDims()