Searched defs:DimIndex (Results 1 - 2 of 2) sorted by relevance
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
H A D | TensorConvolution.h | 490 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 D | dnn.h | 63 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