Searched refs:tensor_map (Results 1 – 3 of 3) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_scan.cpp | 89 TensorMap<Tensor<int, 1, DataLayout> > tensor_map(inputs, 20); in test_tensor_maps() local
90 tensor_map.setRandom(); in test_tensor_maps()
92 Tensor<int, 1, DataLayout> result = tensor_map.cumsum(0); in test_tensor_maps()
96 accum += tensor_map(i); in test_tensor_maps()
|
| D | cxx11_tensor_io.cpp | 112 TensorMap<Tensor<const int, 1, DataLayout> > tensor_map(tensor.data(), 5); in test_output_const() local
115 os << tensor_map; in test_output_const()
|
| D | cxx11_tensor_reduction.cpp | 334 TensorMap<Tensor<int, 4, DataLayout> > tensor_map(inputs, 2, 3, 5, 7); in test_tensor_maps() local
340 tensor_map.setRandom(); in test_tensor_maps()
345 Tensor<int, 2, DataLayout> result = tensor_map.sum(reduction_axis); in test_tensor_maps()
355 sum += tensor_map(i, k, j, l); in test_tensor_maps()
|