Searched refs:TensorIndex (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | eigen_backward_cuboid_convolutions.h | 94 typedef typename internal::traits<OutputBackward>::Index TensorIndex; typedef 97 internal::traits<Kernel>::Layout, TensorIndex>> 102 internal::traits<OutputBackward>::Layout, TensorIndex>> 116 const TensorIndex kernelFilters = 119 const TensorIndex kernelChannels = 121 const TensorIndex kernelPlanes = 123 const TensorIndex kernelRows = 125 const TensorIndex kernelCols = 128 const TensorIndex outputPlanes = 130 const TensorIndex outputRows = [all …]
|
D | eigen_backward_spatial_convolutions.h | 104 typedef typename internal::traits<OutputBackward>::Index TensorIndex; typedef 108 internal::traits<Kernel>::Layout, TensorIndex> > 111 internal::traits<OutputBackward>::Layout, TensorIndex> > 125 const TensorIndex kernelFilters = 128 const TensorIndex kernelChannels = 130 const TensorIndex kernelRows = 132 const TensorIndex kernelCols = 138 const TensorIndex kernelRowsEff = 140 const TensorIndex kernelColsEff = 143 const TensorIndex outputRows = isColMajor [all …]
|
D | eigen_pooling.h | 71 typedef typename internal::traits<Input>::Index TensorIndex; typedef 74 internal::traits<Input>::Layout, TensorIndex> > 92 Eigen::DSizes<TensorIndex, internal::traits<Input>::NumDimensions> 188 typedef typename internal::traits<Input>::Index TensorIndex; in CuboidMaxPooling() typedef 191 internal::traits<Input>::Layout, TensorIndex> > in CuboidMaxPooling() 428 typedef typename internal::traits<Input>::Index TensorIndex; 431 internal::traits<Input>::Layout, TensorIndex> > 449 Eigen::DSizes<TensorIndex, internal::traits<Input>::NumDimensions> 544 typedef typename internal::traits<Input>::Index TensorIndex; 547 internal::traits<Input>::Layout, TensorIndex> >
|
D | eigen_cuboid_convolution.h | 1650 typedef typename internal::traits<Input>::Index TensorIndex; 1653 internal::traits<Input>::Layout, TensorIndex> > 1657 internal::traits<Kernel>::Layout, TensorIndex> > 1668 const TensorIndex kernelFilters = 1670 const TensorIndex kernelChannels = 1674 const TensorIndex kernelPlanes = 1676 const TensorIndex kernelRows = 1678 const TensorIndex kernelCols = 1687 const TensorIndex inputPlanes = 1689 const TensorIndex inputRows = [all …]
|
D | eigen_spatial_convolutions-inl.h | 1364 typedef typename internal::traits<Input>::Index TensorIndex; 1367 internal::traits<Input>::Layout, TensorIndex> > 1371 internal::traits<Kernel>::Layout, TensorIndex> > 1383 const TensorIndex kernelFilters = 1386 const TensorIndex kernelChannels = 1388 const TensorIndex kernelRows = 1390 const TensorIndex kernelCols = 1398 array<IndexPair<TensorIndex>, 1> contract_dims; 1399 contract_dims[0] = IndexPair<TensorIndex>(1, 0); 1401 const TensorIndex InputRows = [all …]
|
/external/eigen/bench/tensors/ |
D | tensor_benchmarks.h | 4 typedef int TensorIndex; typedef 48 Eigen::array<TensorIndex, 2> sizes; in typeCasting() 56 const TensorMap<Tensor<int, 2, 0, TensorIndex>, Eigen::Aligned> A((int*)a_, sizes); in typeCasting() 57 TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, sizes); in typeCasting() 69 Eigen::array<TensorIndex, 2> sizes; in random() 84 Eigen::array<TensorIndex, 2> sizes; in slicing() 91 const Eigen::DSizes<TensorIndex, 2> quarter_sizes(m_/2, m_/2); in slicing() 92 const Eigen::DSizes<TensorIndex, 2> first_quadrant(0, 0); in slicing() 93 const Eigen::DSizes<TensorIndex, 2> second_quadrant(0, m_/2); in slicing() 94 const Eigen::DSizes<TensorIndex, 2> third_quadrant(m_/2, 0); in slicing() [all …]
|
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
D | eigen_spatial_convolutions.h | 103 typedef typename internal::traits<Input>::Index TensorIndex; typedef 106 internal::traits<Input>::Layout, TensorIndex> > 110 internal::traits<Kernel>::Layout, TensorIndex> > 122 const TensorIndex kernelFilters = 125 const TensorIndex kernelChannels = 127 const TensorIndex kernelRows = 129 const TensorIndex kernelCols = 137 array<IndexPair<TensorIndex>, 1> contract_dims; 138 contract_dims[0] = IndexPair<TensorIndex>(1, 0); 140 const TensorIndex InputRows = [all …]
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | eigen_spatial_convolutions.h | 100 typedef typename internal::traits<Input>::Index TensorIndex; typedef 103 internal::traits<Input>::Layout, TensorIndex> > 107 internal::traits<Kernel>::Layout, TensorIndex> > 119 const TensorIndex kernelFilters = 122 const TensorIndex kernelChannels = 124 const TensorIndex kernelRows = 126 const TensorIndex kernelCols = 134 array<IndexPair<TensorIndex>, 1> contract_dims; 135 contract_dims[0] = IndexPair<TensorIndex>(1, 0); 137 const TensorIndex InputRows = [all …]
|