Home
last modified time | relevance | path

Searched refs:TensorIndex (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h105 typedef typename internal::traits<OutputBackward>::Index TensorIndex; typedef
109 internal::traits<Kernel>::Layout, TensorIndex>>
112 internal::traits<OutputBackward>::Layout, TensorIndex>>
126 const TensorIndex kernelFilters =
129 const TensorIndex kernelChannels =
131 const TensorIndex kernelRows =
133 const TensorIndex kernelCols =
139 const TensorIndex kernelRowsEff =
141 const TensorIndex kernelColsEff =
144 const TensorIndex outputRows = isColMajor
[all …]
Deigen_backward_cuboid_convolutions.h94 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 …]
Deigen_pooling.h71 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()
431 typedef typename internal::traits<Input>::Index TensorIndex;
434 internal::traits<Input>::Layout, TensorIndex> >
452 Eigen::DSizes<TensorIndex, internal::traits<Input>::NumDimensions>
547 typedef typename internal::traits<Input>::Index TensorIndex;
550 internal::traits<Input>::Layout, TensorIndex> >
Deigen_cuboid_convolution.h1844 typedef typename internal::traits<Input>::Index TensorIndex;
1847 internal::traits<Input>::Layout, TensorIndex> >
1851 internal::traits<Kernel>::Layout, TensorIndex> >
1862 const TensorIndex kernelFilters =
1864 const TensorIndex kernelChannels =
1868 const TensorIndex kernelPlanes =
1870 const TensorIndex kernelRows =
1872 const TensorIndex kernelCols =
1881 const TensorIndex inputPlanes =
1883 const TensorIndex inputRows =
[all …]
Deigen_spatial_convolutions-inl.h1607 typedef typename internal::traits<Input>::Index TensorIndex;
1610 internal::traits<Input>::Layout, TensorIndex> >
1614 internal::traits<Kernel>::Layout, TensorIndex> >
1626 const TensorIndex kernelFilters =
1629 const TensorIndex kernelChannels =
1631 const TensorIndex kernelRows =
1633 const TensorIndex kernelCols =
1641 array<IndexPair<TensorIndex>, 1> contract_dims;
1642 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
1644 const TensorIndex InputRows =
[all …]
/external/eigen/bench/tensors/
Dtensor_benchmarks.h4 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/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h100 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 …]