Searched refs:float_data_ptr (Results 1 – 2 of 2) sorted by relevance
62 float* float_data_ptr(const at::Tensor& t) { in float_data_ptr() function145 auto x_ptr = float_data_ptr(x); in execute()146 auto h_ptr = float_data_ptr(h); in execute()147 auto y_ptr = float_data_ptr(y); in execute()
153 inline void BatchQuantizeFloats(const float* float_data_ptr, int n_batch, in BatchQuantizeFloats() argument161 float_data_ptr + offset, n_data, quantized_data_ptr + offset, in BatchQuantizeFloats()166 float_data_ptr + offset, n_data, quantized_data_ptr + offset, in BatchQuantizeFloats()