Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
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_backward_spatial_convolutions.h104 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 …]
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()
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> >
Deigen_cuboid_convolution.h1650 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 …]
Deigen_spatial_convolutions-inl.h1364 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/
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/contrib/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h103 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/
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 …]