Searched defs:softmaxstats (Results 1 – 1 of 1) sorted by relevance
25 Tensor& softmaxstats, in run_cudnn_SDP_fprop()48 const Tensor& softmaxstats, in run_cudnn_SDP_bprop()268 Tensor& softmaxstats, in build_graph_and_tensors()404 const Tensor& softmaxstats, in build_graph_and_tensors_backward()535 Tensor& softmaxstats, in run_cudnn_SDP_fprop()624 const Tensor& softmaxstats, in run_cudnn_SDP_bprop()