Searched defs:input_data_base (Results 1 – 5 of 5) sorted by relevance
35 const scalar_t* input_data_base, in _vec_log_softmax_lastdim()115 const scalar_t* input_data_base, in _vec_softmax_lastdim()149 const scalar_t* input_data_base, in _vec_softmax_lastdim()677 const scalar_t* input_data_base = input.const_data_ptr<scalar_t>(); in apply() local692 const scalar_t* input_data_base, in _vec_softmax()800 const scalar_t* input_data_base, in _vec_softmax()894 const scalar_t* input_data_base, in _vec_logsoftmax()999 const scalar_t* input_data_base, in _vec_logsoftmax()1135 const scalar_t* input_data_base = input.const_data_ptr<scalar_t>(); in apply() local
57 const CTYPE_IN* input_data_base = in.const_data_ptr<CTYPE_IN>(); in mul_tensors() local94 const CTYPE_IN* input_data_base = in.const_data_ptr<CTYPE_IN>(); in slice_tensor() local
48 const char* const input_data_base = in cumsum_tensors() local
33 const IN_T* __restrict__ input_data_base = input.const_data_ptr<IN_T>(); in log_softmax_kernel() local
178 scalar_t* input_data_base = input.data_ptr<scalar_t>(); local