Home
last modified time | relevance | path

Searched defs:saved_var (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DNormalization.cpp164 ideep::tensor saved_var; in mkldnn_batch_norm() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc1551 void* saved_mean, void* saved_var, in SetBatchNormForwardArgs()
1652 const void* saved_var, in SetBatchNormBackwardArgs()