Home
last modified time | relevance | path

Searched refs:output_batch_descriptor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h541 dnn::BatchDescriptor* output_batch_descriptor);
Dcuda_dnn.cc4093 dnn::BatchDescriptor* output_batch_descriptor) { in DeriveOutputBatchDescriptor() argument
4103 output_batch_descriptor->set_count(dims[0]) in DeriveOutputBatchDescriptor()
4108 output_batch_descriptor->set_spatial_dim(static_cast<dnn::DimIndex>(i), in DeriveOutputBatchDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h524 dnn::BatchDescriptor* output_batch_descriptor);
Drocm_dnn.cc3990 dnn::BatchDescriptor* output_batch_descriptor) { in DeriveOutputBatchDescriptor() argument
4007 output_batch_descriptor->set_count(dims[0]) in DeriveOutputBatchDescriptor()
4012 output_batch_descriptor->set_spatial_dim(static_cast<dnn::DimIndex>(i), in DeriveOutputBatchDescriptor()