Searched refs:DimIndex (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 55 enum class DimIndex : int { enum 62 inline int64 GetDim(absl::Span<const int64> data, DimIndex dim) { in GetDim() 66 inline void SetDim(absl::Span<int64> data, DimIndex dim, int64 value) { in SetDim() 70 inline void SetDim(std::vector<int64>* data, DimIndex dim, int64 value) { in SetDim() 252 int64 height() const { return GetDim(spatial_size(), DimIndex::Y); } in height() 253 int64 width() const { return GetDim(spatial_size(), DimIndex::X); } in width() 254 int64 spatial_dim(DimIndex dim) const { return GetDim(spatial_size(), dim); } in spatial_dim() 280 SetDim(spatial_size(), DimIndex::Y, value); in set_height() 284 SetDim(spatial_size(), DimIndex::X, value); in set_width() 287 BatchDescriptor& set_spatial_dim(DimIndex dim, int64 value) { in set_spatial_dim() [all …]
|
/external/tensorflow/tensorflow/core/kernels/ |
D | conv_ops_3d.cc | 39 using stream_executor::dnn::DimIndex; 400 .set_spatial_dim(DimIndex::X, in_cols) in launch() 401 .set_spatial_dim(DimIndex::Y, in_rows) in launch() 402 .set_spatial_dim(DimIndex::Z, in_planes) in launch() 406 .set_spatial_dim(DimIndex::X, out_cols) in launch() 407 .set_spatial_dim(DimIndex::Y, out_rows) in launch() 408 .set_spatial_dim(DimIndex::Z, out_planes) in launch() 412 filter_desc.set_spatial_dim(DimIndex::X, filter_cols) in launch() 413 .set_spatial_dim(DimIndex::Y, filter_rows) in launch() 414 .set_spatial_dim(DimIndex::Z, filter_planes) in launch() [all …]
|
D | conv_grad_ops_3d.cc | 45 using stream_executor::dnn::DimIndex; 1300 .set_spatial_dim(DimIndex::X, compatible_input_shape.dim_size(4)) in Compute() 1301 .set_spatial_dim(DimIndex::Y, compatible_input_shape.dim_size(3)) in Compute() 1302 .set_spatial_dim(DimIndex::Z, compatible_input_shape.dim_size(2)) in Compute() 1307 .set_spatial_dim(DimIndex::X, dims.output_size(2)) in Compute() 1308 .set_spatial_dim(DimIndex::Y, dims.output_size(1)) in Compute() 1309 .set_spatial_dim(DimIndex::Z, dims.output_size(0)) in Compute() 1313 filter_desc.set_spatial_dim(DimIndex::X, dims.filter_size(2)) in Compute() 1314 .set_spatial_dim(DimIndex::Y, dims.filter_size(1)) in Compute() 1315 .set_spatial_dim(DimIndex::Z, dims.filter_size(0)) in Compute() [all …]
|
D | cudnn_pooling_gpu.cc | 82 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() 218 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_conv_runner.cc | 38 using se::dnn::DimIndex; 383 static_cast<DimIndex>(effective_num_dimensions - dim - 1), in GetGpuConvConfig() 396 static_cast<DimIndex>(effective_num_dimensions - dim - 1), in GetGpuConvConfig() 406 static_cast<DimIndex>(effective_num_dimensions - dim - 1), in GetGpuConvConfig() 409 static_cast<DimIndex>(effective_num_dimensions - dim - 1), in GetGpuConvConfig() 412 static_cast<DimIndex>(effective_num_dimensions - dim - 1), in GetGpuConvConfig() 424 static_cast<DimIndex>(effective_num_dimensions - dim - 1), in GetGpuConvConfig() 430 input_descriptor.set_spatial_dim(static_cast<DimIndex>(dim), 1); in GetGpuConvConfig() 431 output_descriptor.set_spatial_dim(static_cast<DimIndex>(dim), 1); in GetGpuConvConfig() 432 filter_descriptor.set_spatial_dim(static_cast<DimIndex>(dim), 1); in GetGpuConvConfig() [all …]
|
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
D | TensorReduction.h | 129 template <int DimIndex, typename Self, typename Op> 132 EIGEN_STATIC_ASSERT((DimIndex > 0), YOU_MADE_A_PROGRAMMING_MISTAKE); 133 for (int j = 0; j < self.m_reducedDims[DimIndex]; ++j) { 134 const typename Self::Index input = firstIndex + j * self.m_reducedStrides[DimIndex]; 135 GenericDimReducer<DimIndex-1, Self, Op>::reduce(self, input, reducer, accum); 183 template <int DimIndex, typename Self, typename Op, bool vectorizable = (Self::InputPacketAccess & … 190 template <int DimIndex, typename Self, typename Op> 191 struct InnerMostDimPreserver<DimIndex, Self, Op, true> { 193 EIGEN_STATIC_ASSERT((DimIndex > 0), YOU_MADE_A_PROGRAMMING_MISTAKE); 194 for (typename Self::Index j = 0; j < self.m_reducedDims[DimIndex]; ++j) { [all …]
|
D | TensorConvolution.h | 490 …EIGEN_DEVICE_FUNC void convolve(Index firstIndex, Index firstKernel, int DimIndex, CoeffReturnType… 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& a… 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/cuda/ |
D | cuda_dnn.cc | 4641 output_batch_descriptor->set_spatial_dim(static_cast<dnn::DimIndex>(i), in DeriveOutputBatchDescriptor()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.cc | 4608 output_batch_descriptor->set_spatial_dim(static_cast<dnn::DimIndex>(i), in DeriveOutputBatchDescriptor()
|