Home
last modified time | relevance | path

Searched defs:batch_descriptor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.h390 Stream* stream, const dnn::BatchDescriptor& batch_descriptor, in DoSeparableConvolve()
H A Dcuda_dnn.cc614 CudnnTensorDescriptor(const dnn::BatchDescriptor& batch_descriptor, in CudnnTensorDescriptor()
6534 const dnn::BatchDescriptor& batch_descriptor, in DeriveOutputBatchDescriptor()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.h394 Stream* stream, const dnn::BatchDescriptor& batch_descriptor, in DoSeparableConvolve()
H A Drocm_dnn.cc651 ScopedTensorDescriptor(const BatchDescriptor& batch_descriptor, in ScopedTensorDescriptor()
4524 const BatchDescriptor& batch_descriptor, in DeriveOutputBatchDescriptor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h1322 const BatchDescriptor& batch_descriptor, in PrepareForConvolution()
2621 const BatchDescriptor& batch_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution()
H A Dstream.cc512 const dnn::BatchDescriptor &batch_descriptor, in ThenSeparableConvolve()