Searched defs:scale_offset_mean_variance_descriptor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1402 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationInference() 1472 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in GetFusionOpHashValue() 1502 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationForward() 1572 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in GetFusionOpHashValue() 1603 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in ScopedFusionPlanBatchNormActivationBackward() 1674 miopenTensorDescriptor_t scale_offset_mean_variance_descriptor, in GetFusionOpHashValue() 4668 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationInferenceImpl() 4743 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationInference() 4759 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationInference() 4777 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationForwardImpl() [all …]
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.h | 2464 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationInference() 2477 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationInference() 2518 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationForward() 2531 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationForward() 2578 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationBackward() 2596 const dnn::BatchDescriptor& scale_offset_mean_variance_descriptor, in DoFusedBatchNormActivationBackward()
|