Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc88 typename TTypes<U>::Vec saved_var(saved_var_output->vec<U>()); in operator ()() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1384 void* saved_mean, void* saved_var, in SetBatchNormForwardArgs()
1485 const void* saved_var, in SetBatchNormBackwardArgs()