Searched refs:n_dims (Results 1 – 4 of 4) sorted by relevance
61 auto n_dims = static_cast<int32>(shape.rank()); in GetDiagonalMask() local62 TF_RET_CHECK(n_dims >= 2); in GetDiagonalMask()63 auto m = shape.dimensions(n_dims - 2); in GetDiagonalMask()64 auto n = shape.dimensions(n_dims - 1); in GetDiagonalMask()66 AsInt64Slice(shape.dimensions()).subspan(/*pos=*/0, /*len=*/n_dims - 2); in GetDiagonalMask()79 auto n_dims = static_cast<int32>(shape.rank()); in GetMatrixDiagonal() local80 TF_RET_CHECK(n_dims >= 2); in GetMatrixDiagonal()81 const int64 m = shape.dimensions(n_dims - 2); in GetMatrixDiagonal()82 const int64 n = shape.dimensions(n_dims - 1); in GetMatrixDiagonal()86 zero_size_shape.DeleteDimension(n_dims - 1); in GetMatrixDiagonal()[all …]
50 const int64 n_dims = shape.rank(); in SliceInMinorDims() local51 TF_RET_CHECK(n_minor_dims <= n_dims); in SliceInMinorDims()55 /*len=*/n_dims - n_minor_dims); in SliceInMinorDims()58 std::vector<int64> padded_start(n_dims, 0); in SliceInMinorDims()63 std::vector<int64> padded_end(n_dims); in SliceInMinorDims()67 std::vector<int64> strides(n_dims, 1); in SliceInMinorDims()76 const int64 n_dims = shape.rank(); in UpdateSlice() local78 TF_RET_CHECK(start_size == n_dims); in UpdateSlice()95 const int64 n_dims = shape.rank(); in UpdateSliceInMinorDims() local97 TF_RET_CHECK(n_minor_dims <= n_dims); in UpdateSliceInMinorDims()[all …]
4 const int n_dims = 5; variable6 MatrixType m1(n_dims), m2(n_dims);
1340 int n_dims; in CheckAndFetchProjectionWeights() local1345 /*nbDims=*/&n_dims, /*filterDimA=*/dims)); in CheckAndFetchProjectionWeights()1430 int n_dims; in Create() local1435 /*nbDims=*/&n_dims, /*filterDimA=*/dims)); in Create()