| /external/eigen/doc/snippets/ |
| Tutorial_Map_using.cpp | 4 const int n_dims = 5; variable
6 MatrixType m1(n_dims), m2(n_dims);
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| slicing.cc | 30 const int64 n_dims = shape.rank(); local
31 TF_RET_CHECK(n_minor_dims <= n_dims);
35 /*len=*/n_dims - n_minor_dims);
38 std::vector<int64> padded_start(n_dims, 0);
43 std::vector<int64> padded_end(n_dims);
47 std::vector<int64> strides(n_dims, 1);
56 const int64 n_dims = shape.rank(); local
57 TF_RET_CHECK(start.size() == n_dims);
74 const int64 n_dims = shape.rank();
76 TF_RET_CHECK(n_minor_dims <= n_dims);
[all...] |
| matrix.cc | 53 const int64 n_dims = shape.rank(); local
54 TF_RET_CHECK(n_dims >= 2);
55 const int64 m = shape.dimensions(n_dims - 2);
56 const int64 n = shape.dimensions(n_dims - 1);
61 AsInt64Slice(shape.dimensions()).subspan(/*pos=*/0, /*len=*/n_dims - 2);
76 reducer, {m >= n ? n_dims - 2 : n_dims - 1});
79 ScalarLike(x, 0), reducer, {n_dims - 2});
84 ScalarLike(x, 0), reducer, {n_dims - 1});
95 const int64 n_dims = shape.rank() local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| cholesky_expander.cc | 58 const int n_dims = a_shape.rank(); local
63 /*len=*/n_dims - 2);
87 /*iota_dimension=*/n_dims - 1);
90 /*iota_dimension=*/n_dims - 2);
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_dnn.cc | 1230 int n_dims; local
[all...] |
| /cts/apps/CtsVerifier/libs/ |
| opencv3-android.jar | |
