Searched refs:space_tensor_ptr (Results 1 – 2 of 2) sorted by relevance
39 static void run(T* space_tensor_ptr, const int64* space_tensor_shape, in run()50 space_tensor_ptr + space_tensor_pos * space_tensor_strides[0], in run()70 static void run(T* space_tensor_ptr, const int64* space_tensor_shape, in run()77 batch_tensor_ptr[i] = space_tensor_ptr[i]; in run()79 space_tensor_ptr[i] = batch_tensor_ptr[i]; in run()127 T* space_tensor_ptr = const_cast<T*>(space_tensor.data()); in operator ()() local144 space_tensor_ptr + space_tensor_b * space_tensor_strides[0], in operator ()()
47 __global__ void S2B(const int32 nthreads, T* __restrict__ space_tensor_ptr, in S2B() argument89 ldg(space_tensor_ptr + space_tensor_idx); in S2B()91 space_tensor_ptr[space_tensor_idx] = in S2B()