OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:TensorIndex
(Results
1 - 6
of
6
) sorted by null
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
eigen_spatial_convolutions.h
103
typedef typename internal::traits<Input>::Index
TensorIndex
;
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/core/kernels/
eigen_cuboid_convolution.h
77
typedef typename internal::traits<Input>::Index
TensorIndex
;
80
internal::traits<Input>::Layout,
TensorIndex
> >
84
internal::traits<Kernel>::Layout,
TensorIndex
> >
95
const
TensorIndex
kernelFilters =
97
const
TensorIndex
kernelChannels =
101
const
TensorIndex
kernelDepth =
103
const
TensorIndex
kernelRows =
105
const
TensorIndex
kernelCols =
114
const
TensorIndex
inputPlanes =
116
const
TensorIndex
inputRows
[
all
...]
eigen_backward_spatial_convolutions.h
104
typedef typename internal::traits<OutputBackward>::Index
TensorIndex
;
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 = isColMajo
[
all
...]
eigen_pooling.h
71
typedef typename internal::traits<Input>::Index
TensorIndex
;
74
internal::traits<Input>::Layout,
TensorIndex
> >
92
Eigen::DSizes<
TensorIndex
, internal::traits<Input>::NumDimensions>
188
typedef typename internal::traits<Input>::Index
TensorIndex
;
191
internal::traits<Input>::Layout,
TensorIndex
> >
420
typedef typename internal::traits<Input>::Index
TensorIndex
;
423
internal::traits<Input>::Layout,
TensorIndex
> >
441
Eigen::DSizes<
TensorIndex
, internal::traits<Input>::NumDimensions>
538
typedef typename internal::traits<Input>::Index
TensorIndex
;
541
internal::traits<Input>::Layout,
TensorIndex
> >
[
all
...]
eigen_backward_cuboid_convolutions.h
87
typedef typename internal::traits<OutputBackward>::Index
TensorIndex
;
90
internal::traits<Kernel>::Layout,
TensorIndex
> >
95
internal::traits<OutputBackward>::Layout,
TensorIndex
> >
109
const
TensorIndex
kernelFilters =
112
const
TensorIndex
kernelChannels =
114
const
TensorIndex
kernelPlanes =
116
const
TensorIndex
kernelRows =
118
const
TensorIndex
kernelCols =
121
const
TensorIndex
outputPlanes =
123
const
TensorIndex
outputRows
[
all
...]
/external/eigen/bench/tensors/
tensor_benchmarks.h
4
typedef int
TensorIndex
;
48
Eigen::array<
TensorIndex
, 2> sizes;
56
const TensorMap<Tensor<int, 2, 0,
TensorIndex
>, Eigen::Aligned> A((int*)a_, sizes);
57
TensorMap<Tensor<T, 2, 0,
TensorIndex
>, Eigen::Aligned> B(b_, sizes);
69
Eigen::array<
TensorIndex
, 2> sizes;
84
Eigen::array<
TensorIndex
, 2> sizes;
91
const Eigen::DSizes<
TensorIndex
, 2> quarter_sizes(m_/2, m_/2);
92
const Eigen::DSizes<
TensorIndex
, 2> first_quadrant(0, 0);
93
const Eigen::DSizes<
TensorIndex
, 2> second_quadrant(0, m_/2);
94
const Eigen::DSizes<
TensorIndex
, 2> third_quadrant(m_/2, 0)
[
all
...]
Completed in 242 milliseconds