Searched defs:output_data_base (Results 1 – 5 of 5) sorted by relevance
36 scalar_t* output_data_base, in _vec_log_softmax_lastdim()116 scalar_t* output_data_base, in _vec_softmax_lastdim()150 scalar_t* output_data_base, in _vec_softmax_lastdim()216 const scalar_t* output_data_base, in _vec_host_softmax_backward_lastdim()267 const scalar_t* output_data_base, in _vec_softmax_backward()351 const scalar_t* output_data_base, in _vec_softmax_backward()478 const scalar_t* output_data_base, in _vec_log_softmax_backward()561 const scalar_t* output_data_base, in _vec_log_softmax_backward()678 scalar_t* output_data_base = output.data_ptr<scalar_t>(); in apply() local693 scalar_t* output_data_base, in _vec_softmax()[all …]
58 CTYPE_OUT* output_data_base = out.mutable_data_ptr<CTYPE_OUT>(); in mul_tensors() local95 CTYPE_OUT* output_data_base = out.mutable_data_ptr<CTYPE_OUT>(); in slice_tensor() local
50 CTYPE_OUT* output_data_base = out.mutable_data_ptr<CTYPE_OUT>(); in cumsum_tensors() local
179 scalar_t* output_data_base = output.data_ptr<scalar_t>(); local286 scalar_t* output_data_base = output.data_ptr<scalar_t>(); in host_softmax_backward() local
34 OUT_T* __restrict__ output_data_base = out.mutable_data_ptr<OUT_T>(); in log_softmax_kernel() local