Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1697 const CudnnTensorDescriptor& x_descriptor, in CreateBatchNormForwardWorkspace() argument
1705 /*xDesc=*/x_descriptor.handle(), /*zDesc=*/x_descriptor.handle(), in CreateBatchNormForwardWorkspace()
1706 /*yDesc=*/x_descriptor.handle(), in CreateBatchNormForwardWorkspace()
1720 const CudnnTensorDescriptor& x_descriptor, in CreateBatchNormBackwardWorkspace() argument
1727 /*xDesc=*/x_descriptor.handle(), in CreateBatchNormBackwardWorkspace()
1728 /*yDesc=*/x_descriptor.handle(), in CreateBatchNormBackwardWorkspace()
1729 /*dyDesc=*/x_descriptor.handle(), in CreateBatchNormBackwardWorkspace()
1731 /*dxDesc=*/x_descriptor.handle(), in CreateBatchNormBackwardWorkspace()
3508 CudnnTensorDescriptor x_descriptor(x_desc, ToCudnnDataType(input_data_type)); in DoBatchNormalizationForwardImpl() local
3542 stream, cudnn, mode, bn_ops, activation_desc.handle(), x_descriptor, in DoBatchNormalizationForwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h598 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
609 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
620 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
631 Stream* stream, const dnn::BatchDescriptor& x_descriptor,
797 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data,
808 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data,
Drocm_dnn.cc3577 ScopedTensorDescriptor x_descriptor{x_desc, in DoBatchNormalizationForwardImpl() local
3588 miopen.handle(), mode, &one, &zero, x_descriptor.handle(), x.opaque(), in DoBatchNormalizationForwardImpl()
3589 x_descriptor.handle(), y->opaque(), scale_offset_descriptor.handle(), in DoBatchNormalizationForwardImpl()
3596 miopen.handle(), mode, &one, &zero, x_descriptor.handle(), x.opaque(), in DoBatchNormalizationForwardImpl()
3597 x_descriptor.handle(), y->opaque(), scale_offset_descriptor.handle(), in DoBatchNormalizationForwardImpl()
3652 ScopedTensorDescriptor x_descriptor{ in DoBatchNormalizationBackwardImpl() local
3661 miopen.handle(), mode, &one, &zero, &one, &zero, x_descriptor.handle(), in DoBatchNormalizationBackwardImpl()
3662 x.opaque(), x_descriptor.handle(), y_backprop.opaque(), in DoBatchNormalizationBackwardImpl()
3663 x_descriptor.handle(), x_backprop->opaque(), in DoBatchNormalizationBackwardImpl()
4728 const dnn::BatchDescriptor& x_descriptor, const DeviceMemory<T>& x_data, in DoFusedBatchNormActivationInferenceImpl() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2528 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference() argument
2541 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationInference() argument
2582 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward() argument
2595 Stream* stream, const dnn::BatchDescriptor& x_descriptor, in DoFusedBatchNormActivationForward() argument