Searched refs:dim4 (Results 1 – 5 of 5) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | Tensor.h | 363 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2, Index dim3, Index dim4) in Tensor() argument
364 : m_storage(dim1*dim2*dim3*dim4, array<Index, 4>(dim1, dim2, dim3, dim4)) in Tensor()
368 …EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Tensor(Index dim1, Index dim2, Index dim3, Index dim4, Index… in Tensor() argument
369 : m_storage(dim1*dim2*dim3*dim4*dim5, array<Index, 5>(dim1, dim2, dim3, dim4, dim5)) in Tensor()
|
| D | TensorMap.h | 86 …tr, Index dim1, Index dim2, Index dim3, Index dim4) : m_data(dataPtr), m_dimensions(dim1, dim2, di… in TensorMap() argument
90 …dex dim1, Index dim2, Index dim3, Index dim4, Index dim5) : m_data(dataPtr), m_dimensions(dim1, di… in TensorMap() argument
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_index_list.cpp | 305 int dim4 = 3; in test_mixed_index_list() local
307 auto reduction_axis = make_index_list(0, dim2, 2, dim4); in test_mixed_index_list()
|
| /external/llvm-project/mlir/test/Dialect/Linalg/ |
| D | loops.mlir | 1336 // CHECKLOOP: %[[dim4:.*]] = dim %[[arg2]], %[[c1]] : memref<?x?x?xf32>
1339 // CHECKLOOP: scf.for %[[arg4:.*]] = %[[c0]] to %[[dim4]] step %[[c1]] {
1367 // CHECKPARALLEL: %[[dim4:.*]] = dim %[[arg2]], %[[c1]] : memref<?x?x?xf32>
1369 …[arg4:.*]], %[[arg5:.*]]) = (%[[c0]], %[[c0]], %[[c0]]) to (%[[dim3]], %[[dim4]], %[[dim5]]) step …
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 188 void DescribeTensor5D(int dim0, int dim1, int dim2, int dim3, int dim4, in DescribeTensor5D() argument
195 shape->add_dim()->set_size(dim4); in DescribeTensor5D()
|