Searched refs:num_out_dims (Results 1 – 2 of 2) sorted by relevance
1048 gtl::ArraySlice<int64> orig, int64 num_out_dims) { in ComputeFlatInnerDims() argument1049 gtl::InlinedVector<int64, 4> out_dims(num_out_dims, 0); in ComputeFlatInnerDims()1050 int64 offset = orig.size() - num_out_dims; in ComputeFlatInnerDims()1051 for (int64 out_dim = num_out_dims - 1; out_dim >= 0; --out_dim) { in ComputeFlatInnerDims()1062 gtl::ArraySlice<int64> orig, int64 num_out_dims) { in ComputeFlatOuterDims() argument1063 gtl::InlinedVector<int64, 4> out_dims(num_out_dims, 0); in ComputeFlatOuterDims()1064 for (int64 out_dim = 0; out_dim <= num_out_dims - 1; ++out_dim) { in ComputeFlatOuterDims()1067 for (int64 in_dim = num_out_dims; in_dim < orig.size(); ++in_dim) { in ComputeFlatOuterDims()1068 out_dims[num_out_dims - 1] *= orig[in_dim]; in ComputeFlatOuterDims()
471 gtl::ArraySlice<int64> orig, int64 num_out_dims);473 gtl::ArraySlice<int64> orig, int64 num_out_dims);