Searched defs:grad_input_data_base (Results 1 – 1 of 1) sorted by relevance
214 scalar_t* grad_input_data_base, in _vec_host_softmax_backward_lastdim()265 scalar_t* grad_input_data_base, in _vec_softmax_backward()349 scalar_t* grad_input_data_base, in _vec_softmax_backward()476 scalar_t* grad_input_data_base, in _vec_log_softmax_backward()559 scalar_t* grad_input_data_base, in _vec_log_softmax_backward()1155 scalar_t* grad_input_data_base = grad_input.mutable_data_ptr<scalar_t>(); in apply() local1183 scalar_t* grad_input_data_base = grad_input.mutable_data_ptr<scalar_t>(); in apply() local