Searched defs:max_rank (Results 1 – 9 of 9) sorted by relevance
68 int max_rank = std::max(lhs_shape.dims(), rhs_shape.dims()); in Compile() local
69 Output max_rank = ops::Maximum(scope, rank_0, rank_1); in ExpandDimsForBroadcast() local
80 WeightType max_rank; member
320 float max_rank = summary_values(i, 3); in Compute() local
203 int max_rank = 1; in CwiseOutputElementCount() local
2338 const int64 max_rank = 4; in PropagateShapes() local
65 int64_t max_rank = -1; in VerifyOperandsHaveSameShapesOrBroadcastableShape() local
444 std::vector<int64> OpTest::RandomDims(int min_rank, int max_rank, in RandomDims()
318 const int max_rank = nvinfer1::Dims::MAX_DIMS + (use_implicit_batch ? 1 : 0); in ValidateTensorProperties() local