Searched defs:DimIndex (Results 1 - 2 of 2) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorConvolution.h490 EIGEN_DEVICE_FUNC void convolve(Index firstIndex, Index firstKernel, int DimIndex, CoeffReturnType& accum) const { argument
491 for (int j = 0; j < m_kernelImpl.dimensions()[DimIndex]; ++j) {
492 const Index input = firstIndex + j * m_indexStride[DimIndex];
493 const Index kernel = firstKernel + j * m_kernelStride[DimIndex];
494 if (DimIndex > 0) {
495 convolve(input, kernel, DimIndex-1, accum);
503 EIGEN_DEVICE_FUNC void convolvePacket(Index firstIndex, Index firstKernel, int DimIndex, Packet& accum) const { argument
504 for (int j = 0; j < m_kernelImpl.dimensions()[DimIndex]; ++j) {
505 const Index input = firstIndex + j * m_indexStride[DimIndex];
506 const Index kernel = firstKernel + j * m_kernelStride[DimIndex];
[all...]
/external/tensorflow/tensorflow/stream_executor/
H A Ddnn.h63 enum class DimIndex : int { class in namespace:perftools::gputools::dnn
70 inline int64 GetDim(const std::vector<int64>& data, DimIndex dim) {
74 inline void SetDim(std::vector<int64>* data, DimIndex dim, int64 value) {
248 int64 height() const { return GetDim(spatial_size_, DimIndex::Y); }
249 int64 width() const { return GetDim(spatial_size_, DimIndex::X); }
250 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size_, dim); }
276 SetDim(&spatial_size_, DimIndex::Y, value);
280 SetDim(&spatial_size_, DimIndex::X, value);
283 BatchDescriptor& set_spatial_dim(DimIndex dim, int64 value) {
405 SetDim(&input_filter_dims_, DimIndex
[all...]

Completed in 331 milliseconds