Searched defs:min_rank (Results 1 – 8 of 8) sorted by relevance
228 def _get_dimension(shape, dim, min_rank=1): argument256 def channel_dimension(shape, data_format, min_rank=1): argument275 def last_dimension(shape, min_rank=1): argument
22 float min_rank = 3; field
70 int min_rank = std::min(lhs_shape.dims(), rhs_shape.dims()); in Compile() local
78 WeightType min_rank; member
70 WeightType min_rank; member
253 float min_rank = summary_values(i, 2); in Compute() local
1253 const int min_rank = concat_dim < 0 ? -concat_dim : concat_dim + 1; in ConcatShapeHelper() local
443 std::vector<int64> OpTest::RandomDims(int min_rank, int max_rank, in RandomDims()