Searched refs:n_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing.cc | 30 const int64 n_dims = shape.rank(); in SliceInMinorDims() local
31 TF_RET_CHECK(n_minor_dims <= n_dims); in SliceInMinorDims()
35 /*len=*/n_dims - n_minor_dims); in SliceInMinorDims()
38 std::vector<int64> padded_start(n_dims, 0); in SliceInMinorDims()
43 std::vector<int64> padded_end(n_dims); in SliceInMinorDims()
47 std::vector<int64> strides(n_dims, 1); in SliceInMinorDims()
56 const int64 n_dims = shape.rank(); in UpdateSlice() local
57 TF_RET_CHECK(start.size() == n_dims); in UpdateSlice()
74 const int64 n_dims = shape.rank(); in UpdateSliceInMinorDims() local
76 TF_RET_CHECK(n_minor_dims <= n_dims); in UpdateSliceInMinorDims()
[all …]
|
| D | matrix.cc | 53 const int64 n_dims = shape.rank(); in GetMatrixDiagonal() local
54 TF_RET_CHECK(n_dims >= 2); in GetMatrixDiagonal()
55 const int64 m = shape.dimensions(n_dims - 2); in GetMatrixDiagonal()
56 const int64 n = shape.dimensions(n_dims - 1); in GetMatrixDiagonal()
61 AsInt64Slice(shape.dimensions()).subspan(/*pos=*/0, /*len=*/n_dims - 2); in GetMatrixDiagonal()
76 reducer, {m >= n ? n_dims - 2 : n_dims - 1}); in GetMatrixDiagonal()
79 ScalarLike(x, 0), reducer, {n_dims - 2}); in GetMatrixDiagonal()
84 ScalarLike(x, 0), reducer, {n_dims - 1}); in GetMatrixDiagonal()
95 const int64 n_dims = shape.rank(); in TriangleMask() local
96 TF_RET_CHECK(n_dims >= 2); in TriangleMask()
[all …]
|
| /external/eigen/doc/snippets/ |
| D | Tutorial_Map_using.cpp | 4 const int n_dims = 5; variable
6 MatrixType m1(n_dims), m2(n_dims);
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 58 const int n_dims = a_shape.rank(); in CholeskyUnblocked() local
63 /*len=*/n_dims - 2); in CholeskyUnblocked()
87 /*iota_dimension=*/n_dims - 1); in CholeskyUnblocked()
90 /*iota_dimension=*/n_dims - 2); in CholeskyUnblocked()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1230 int n_dims; in Create() local
1235 /*nbDims=*/&n_dims, /*filterDimA=*/dims)); in Create()
|