Searched defs:max_rank (Results 1 – 8 of 8) sorted by relevance
40 def _match_op_rank(self, graph_module, node, arg, max_rank): argument57 def _match_buffer_rank(self, arg, max_rank): argument
15 constexpr int max_rank = 3; variable
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()