Searched refs:x_nd (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 4140 ScopedTensorDescriptor x_nd{x_descriptor, in DoFusedBatchNormActivationInferenceImpl() local
4150 miopen.handle(), x_nd.handle(), scale_offset_mean_variance_nd.handle(), in DoFusedBatchNormActivationInferenceImpl()
4178 status = fusion_plan.Execute(x_nd.handle(), x_data.opaque(), in DoFusedBatchNormActivationInferenceImpl()
4179 x_nd.handle(), y_data->opaque()); in DoFusedBatchNormActivationInferenceImpl()
4251 ScopedTensorDescriptor x_nd{x_descriptor, in DoFusedBatchNormActivationForwardImpl() local
4261 miopen.handle(), x_nd.handle(), scale_offset_mean_variance_nd.handle(), in DoFusedBatchNormActivationForwardImpl()
4290 status = fusion_plan.Execute(x_nd.handle(), x_data.opaque(), in DoFusedBatchNormActivationForwardImpl()
4291 x_nd.handle(), y_data->opaque()); in DoFusedBatchNormActivationForwardImpl()
|