HomeSort by relevance Sort by last modified time
    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