Searched refs:full_dims (Results 1 – 4 of 4) sorted by relevance
279 std::vector<int64> BatchDescriptor::full_dims(const DataLayout& layout) const { in full_dims() function in stream_executor::dnn::BatchDescriptor298 std::vector<int64> phys_dims = full_dims(this->layout()); in full_strides()
264 std::vector<int64> full_dims(const DataLayout& layout) const;
665 batch_descriptor.full_dims(dnn::DataLayout::kBatchDepthYX); in ScopedTensorDescriptor()4017 input_dimensions.full_dims(dnn::DataLayout::kBatchDepthYX) && in IsSame()4019 output_dimensions.full_dims(dnn::DataLayout::kBatchDepthYX) && in IsSame()4062 desc->input_dims = input_dimensions.full_dims(dnn::DataLayout::kBatchDepthYX); in insert()4064 output_dimensions.full_dims(dnn::DataLayout::kBatchDepthYX); in insert()4255 output_dimensions.full_dims(dnn::DataLayout::kBatchDepthYX); in DoPoolBackwardImpl()4433 dimensions.full_dims(dnn::DataLayout::kBatchDepthYX); in DoNormalizeBackwardWithDimensions()
528 batch_descriptor.full_dims(dnn::DataLayout::kBatchDepthYX); in CudnnTensorDescriptor()