Searched refs:num_out_dims (Results 1 – 2 of 2) sorted by relevance
1207 gtl::ArraySlice<int64> orig, int64 num_out_dims) { in ComputeFlatInnerDims() argument1208 gtl::InlinedVector<int64, 4> out_dims(num_out_dims, 0); in ComputeFlatInnerDims()1209 int64 offset = orig.size() - num_out_dims; in ComputeFlatInnerDims()1210 for (int64 out_dim = num_out_dims - 1; out_dim >= 0; --out_dim) { in ComputeFlatInnerDims()1221 gtl::ArraySlice<int64> orig, int64 num_out_dims) { in ComputeFlatOuterDims() argument1222 gtl::InlinedVector<int64, 4> out_dims(num_out_dims, 0); in ComputeFlatOuterDims()1223 for (int64 out_dim = 0; out_dim <= num_out_dims - 1; ++out_dim) { in ComputeFlatOuterDims()1226 for (int64 in_dim = num_out_dims; in_dim < orig.size(); ++in_dim) { in ComputeFlatOuterDims()1227 out_dims[num_out_dims - 1] *= orig[in_dim]; in ComputeFlatOuterDims()
600 gtl::ArraySlice<int64> orig, int64 num_out_dims);602 gtl::ArraySlice<int64> orig, int64 num_out_dims);