Searched refs:full_dims (Results 1 – 4 of 4) sorted by relevance
248 std::vector<int64> BatchDescriptor::full_dims(const DataLayout& layout) const { in full_dims() function in stream_executor::dnn::BatchDescriptor267 std::vector<int64> phys_dims = full_dims(this->layout()); in full_strides()
272 std::vector<int64> full_dims(const DataLayout& layout) const;
549 batch_descriptor.full_dims(dnn::DataLayout::kBatchDepthYX); in ScopedTensorDescriptor()3605 output_dimensions.full_dims(dnn::DataLayout::kBatchDepthYX); in DoPoolBackward()3701 output_dimensions.full_dims(dnn::DataLayout::kBatchDepthYX); in DoPoolBackward()3836 dimensions.full_dims(dnn::DataLayout::kBatchDepthYX); in DoNormalizeBackwardWithDimensions()
506 batch_descriptor.full_dims(dnn::DataLayout::kBatchDepthYX); in CudnnTensorDescriptor()