OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:NumInputDims
(Results
1 - 6
of
6
) sorted by null
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorChipping.h
132
static const int
NumInputDims
= internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
133
static const int NumDims =
NumInputDims
-1;
155
EIGEN_STATIC_ASSERT((
NumInputDims
>= 1), YOU_MADE_A_PROGRAMMING_MISTAKE);
156
eigen_assert(
NumInputDims
> m_dim.actualDim());
162
for (int i = 0; i <
NumInputDims
; ++i) {
177
for (int i =
NumInputDims
-1; i > m_dim.actualDim(); --i) {
209
(static_cast<int>(Layout) == static_cast<int>(RowMajor) && m_dim.actualDim() ==
NumInputDims
-1)) {
220
} else if ((static_cast<int>(Layout) == static_cast<int>(ColMajor) && m_dim.actualDim() ==
NumInputDims
- 1) ||
250
m_dim.actualDim() ==
NumInputDims
- 1)) {
253
m_dim.actualDim() ==
NumInputDims
- 1) |
[
all
...]
TensorReduction.h
61
const int
NumInputDims
= internal::array_size<InputDims>::value;
64
for (int i = 0; i <
NumInputDims
; ++i) {
80
const int
NumInputDims
= internal::array_size<InputDims>::value;
81
for (int i = 0; i <
NumInputDims
; ++i) {
388
static const int
NumInputDims
= internal::array_size<InputDimensions>::value;
390
static const int NumOutputDims =
NumInputDims
- NumReducedDims;
407
static const bool ReducingInnerMostDims = internal::are_inner_most_dims<Dims,
NumInputDims
, Layout>::value;
408
static const bool PreservingInnerMostDims = internal::preserve_inner_most_dims<Dims,
NumInputDims
, Layout>::value;
414
EIGEN_STATIC_ASSERT((
NumInputDims
>= NumReducedDims), YOU_MADE_A_PROGRAMMING_MISTAKE);
415
EIGEN_STATIC_ASSERT((!ReducingInnerMostDims | !PreservingInnerMostDims | (NumReducedDims ==
NumInputDims
)),
[
all
...]
TensorImagePatch.h
155
static const int
NumInputDims
= internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
156
static const int NumDims =
NumInputDims
+ 1;
189
m_inputDepth = input_dims[
NumInputDims
-1];
190
m_inputRows = input_dims[
NumInputDims
-2];
191
m_inputCols = input_dims[
NumInputDims
-3];
271
m_dimensions[NumDims-1] = input_dims[
NumInputDims
-1];
TensorVolumePatch.h
170
static const int
NumInputDims
= internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
171
static const int NumDims =
NumInputDims
+ 1;
203
m_inputDepth = input_dims[
NumInputDims
-1];
204
m_inputPlanes = input_dims[
NumInputDims
-2];
205
m_inputRows = input_dims[
NumInputDims
-3];
206
m_inputCols = input_dims[
NumInputDims
-4];
292
m_dimensions[NumDims-1] = input_dims[
NumInputDims
-1];
/external/tensorflow/tensorflow/core/kernels/
eigen_volume_patch.h
31
static const int
NumInputDims
= internal::array_size<
33
static const int NumDims =
NumInputDims
+ 1;
68
m_inputDepth = input_dims[
NumInputDims
- 1];
69
m_inputPlanes = input_dims[
NumInputDims
- 2];
70
m_inputRows = input_dims[
NumInputDims
- 3];
71
m_inputCols = input_dims[
NumInputDims
- 4];
181
m_dimensions[NumDims - 1] = input_dims[
NumInputDims
- 1];
eigen_spatial_convolutions.h
404
const int
NumInputDims
= array_size<
406
otherIndex = (
NumInputDims
== 3) ? 0 : patchIndex / m_fastNumPatches;
407
const Index patch2DIndex = (
NumInputDims
== 3)
[
all
...]
Completed in 291 milliseconds