Searched refs:num_out_dims (Results 1 – 2 of 2) sorted by relevance
1271 gtl::ArraySlice<int64> orig, int64 num_out_dims) { in ComputeFlatInnerDims() argument1272 gtl::InlinedVector<int64, 4> out_dims(num_out_dims, 0); in ComputeFlatInnerDims()1273 int64 offset = orig.size() - num_out_dims; in ComputeFlatInnerDims()1274 for (int64 out_dim = num_out_dims - 1; out_dim >= 0; --out_dim) { in ComputeFlatInnerDims()1285 gtl::ArraySlice<int64> orig, int64 num_out_dims) { in ComputeFlatOuterDims() argument1286 gtl::InlinedVector<int64, 4> out_dims(num_out_dims, 0); in ComputeFlatOuterDims()1287 for (int64 out_dim = 0; out_dim <= num_out_dims - 1; ++out_dim) { in ComputeFlatOuterDims()1290 for (int64 in_dim = num_out_dims; in_dim < orig.size(); ++in_dim) { in ComputeFlatOuterDims()1291 out_dims[num_out_dims - 1] *= orig[in_dim]; in ComputeFlatOuterDims()
665 gtl::ArraySlice<int64> orig, int64 num_out_dims);667 gtl::ArraySlice<int64> orig, int64 num_out_dims);