Searched defs:n_dims (Results 1 – 5 of 5) sorted by relevance
30 const int64 n_dims = shape.rank(); in SliceInMinorDims() local56 const int64 n_dims = shape.rank(); in UpdateSlice() local74 const int64 n_dims = shape.rank(); in UpdateSliceInMinorDims() local98 const int64 n_dims = shape.rank(); in PrependZerosInMajorDims() local114 const int64 n_dims = shape.rank(); in DynamicSliceInMinorDims() local
53 const int64 n_dims = shape.rank(); in GetMatrixDiagonal() local95 const int64 n_dims = shape.rank(); in TriangleMask() local343 const int64 n_dims = shape.rank(); in TransposeInMinorDims() local
4 const int n_dims = 5; variable
58 const int n_dims = a_shape.rank(); in CholeskyUnblocked() local
1230 int n_dims; in Create() local