Searched refs:ScopedTensorDescriptor (Results 1 – 1 of 1) sorted by relevance
525 class ScopedTensorDescriptor { class527 ScopedTensorDescriptor(const BatchDescriptor& batch_descriptor, in ScopedTensorDescriptor() function in stream_executor::gpu::ScopedTensorDescriptor574 ~ScopedTensorDescriptor() { in ~ScopedTensorDescriptor()587 SE_DISALLOW_COPY_AND_ASSIGN(ScopedTensorDescriptor);2601 ScopedTensorDescriptor input_nd{ in DoPrepareForConvolution()2607 ScopedTensorDescriptor output_nd{ in DoPrepareForConvolution()2787 ScopedTensorDescriptor orig_out_back_nd{ in MaybeTransformLayout()2789 ScopedTensorDescriptor transformed_out_back_nd{ in MaybeTransformLayout()2819 ScopedTensorDescriptor input_nd{input_descriptor, in DoConvolve()2821 ScopedTensorDescriptor output_nd{output_descriptor, in DoConvolve()[all …]