Searched defs:n_dims (Results 1 – 4 of 4) sorted by relevance
50 const int64 n_dims = shape.rank(); in SliceInMinorDims() local76 const int64 n_dims = shape.rank(); in UpdateSlice() local95 const int64 n_dims = shape.rank(); in UpdateSliceInMinorDims() local119 const int64 n_dims = shape.rank(); in PrependZerosInMajorDims() local135 const int64 n_dims = shape.rank(); in DynamicSliceInMinorDims() local
61 auto n_dims = static_cast<int32>(shape.rank()); in GetDiagonalMask() local79 auto n_dims = static_cast<int32>(shape.rank()); in GetMatrixDiagonal() local113 auto n_dims = static_cast<int32>(shape.rank()); in GetMatrixDiagonalViaGather() local177 auto n_dims = static_cast<int32>(shape.rank()); in SetMatrixDiagonal() local215 const int64 n_dims = shape.rank(); in TriangleMask() local701 const int64 n_dims = shape.rank(); in TransposeInMinorDims() local
4 const int n_dims = 5; variable
1340 int n_dims; in CheckAndFetchProjectionWeights() local1430 int n_dims; in Create() local