Searched refs:input_data_base (Results 1 – 6 of 6) sorted by relevance
48 const char* const input_data_base = in cumsum_tensors() local53 output_data_base[0] = load_self(&input_data_base[0]); in cumsum_tensors()66 load_self(&input_data_base[(start_loc + idx) * self.element_size()]); in cumsum_tensors()74 load_self(&input_data_base in cumsum_tensors()
57 const CTYPE_IN* input_data_base = in.const_data_ptr<CTYPE_IN>(); in mul_tensors() local62 input_data_base + i * dim_length_in * trailing_dims; in mul_tensors()94 const CTYPE_IN* input_data_base = in.const_data_ptr<CTYPE_IN>(); in slice_tensor() local98 const CTYPE_IN* input_data = input_data_base + in slice_tensor()
35 const scalar_t* input_data_base, in _vec_log_softmax_lastdim() argument65 const scalar_t* input_data = input_data_base + i * dim_size; in _vec_log_softmax_lastdim()73 const scalar_t* input_data = input_data_base + i * dim_size; in _vec_log_softmax_lastdim()90 const scalar_t* input_data = input_data_base + i * dim_size; in _vec_log_softmax_lastdim()115 const scalar_t* input_data_base, in _vec_softmax_lastdim() argument123 const scalar_t* input_data = input_data_base + i * dim_size; in _vec_softmax_lastdim()149 const scalar_t* input_data_base, in _vec_softmax_lastdim() argument162 const scalar_t* input_data = input_data_base + i * dim_size; in _vec_softmax_lastdim()677 const scalar_t* input_data_base = input.const_data_ptr<scalar_t>(); in apply() local681 input_data_base, output_data_base, outer_size, dim_size); in apply()[all …]
33 const IN_T* __restrict__ input_data_base = input.const_data_ptr<IN_T>(); in log_softmax_kernel() local58 input_data_base + outer_idx * outer_stride + inner_idx; in log_softmax_kernel()
178 scalar_t* input_data_base = input.data_ptr<scalar_t>(); local189 input_data_base + outer_idx * outer_stride + inner_idx; in __anon010efb6d0202()
3366 const int8* input_data_base = input_data_depthwise + 2 * 8 * s;3370 const int8* next_input_data = input_data_base;3783 input_data_base += 4 * workspace_height_stride;3799 const int8* input_data_base = input_data_depthwise;3809 const int8* next_input_data = input_data_base;3903 input_data_base += workspace_height_stride;5764 const int8* input_data_base = input_data_depthwise + 2 * 8 * s;5768 const int8* next_input_data = input_data_base;6186 input_data_base += 4 * workspace_height_stride;6202 const int8* input_data_base = input_data_depthwise;[all …]