Searched refs:swap_layout (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.h | 35 To32Bit(output).swap_layout().device(d) = Eigen::SpatialAvgPooling( in operator()
36 To32Bit(input).swap_layout(), window_cols, window_rows, col_stride, in operator()
40 output.swap_layout().device(d) = Eigen::SpatialAvgPooling( in operator()
41 input.swap_layout(), window_cols, window_rows, col_stride, row_stride, in operator()
|
| D | maxpooling_op.h | 36 output.swap_layout().device(d) = in operator()
37 Eigen::SpatialMaxPooling(input.swap_layout(), window_cols, window_rows, in operator()
|
| D | sparse_tensor_dense_matmul_op.cc | 308 b.swap_layout().shuffle(shuffle).conjugate(); in Compute()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_layout_swap.cpp | 21 Tensor<float, 3, RowMajor> tensor2 = tensor.swap_layout(); in test_simple_swap()
42 tensor2.swap_layout() = tensor; in test_swap_as_lvalue()
|
| D | cxx11_tensor_image_patch.cpp | 20 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout(); in test_simple_patch()
206 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout(); in test_patch_padding_valid()
279 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout(); in test_patch_padding_valid_same_value()
355 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout(); in test_patch_padding_same()
412 Tensor<float, 3, RowMajor> tensor_row_major = tensor.swap_layout(); in test_patch_no_extra_dim()
571 Tensor<float, 5, RowMajor> l_out_row_major = l_in.swap_layout().extract_image_patches(11, 11); in test_imagenet_patches()
621 l_out_row_major = l_in.swap_layout().extract_image_patches(9, 9); in test_imagenet_patches()
667 l_out_row_major = l_in.swap_layout().extract_image_patches(7, 7); in test_imagenet_patches()
713 l_out_row_major = l_in.swap_layout().extract_image_patches(3, 3); in test_imagenet_patches()
|
| D | cxx11_tensor_volume_patch.cpp | 11 Tensor<float, 5, RowMajor> tensor_row_major = tensor.swap_layout(); in test_single_voxel_patch()
49 Tensor<float, 5, RowMajor> tensor_row_major = tensor.swap_layout(); in test_entire_volume_patch()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | attention_ops.cc | 122 output->tensor<float, 4>().swap_layout().device( in Compute()
124 Eigen::ExtractGlimpses(input.tensor<float, 4>().swap_layout(), in Compute()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBase.h | 733 swap_layout() const { in swap_layout() function
885 swap_layout() const { in swap_layout() function
890 swap_layout() { in swap_layout() function
|
| D | README.md | 139 ```swap_layout()``` method. Note that this will also reverse the order of the
149 col_major_result = row_major.swap_layout();
155 col_major_result = row_major.swap_layout().shuffle(shuffle);
1680 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout();
|