Searched refs:b_base_ptr (Results 1 – 3 of 3) sorted by relevance
331 auto* b_base_ptr = in_y.template flat<Scalar>().data();337 b_device_memory.push_back(AsDeviceMemory(b_base_ptr + i * k * n));350 b_device_memory.push_back(AsDeviceMemory(b_base_ptr + i * k * n));467 auto* b_base_ptr = in_y.template flat<Scalar>().data();473 b_device_memory.push_back(AsDeviceMemory(b_base_ptr + i * k * n));486 b_device_memory.push_back(AsDeviceMemory(b_base_ptr + i * k * n));
280 auto* b_base_ptr = in_y.template flat<Scalar>().data();283 b_tmp_ptrs.push_back(b_base_ptr + i * m * n);
547 const T* b_base_ptr = b_t.template flat<T>().data(); in Compute() local562 b_base_ptr = b_conj_t.template flat<T>().data(); in Compute()574 const T* b_i = b_base_ptr + i * b_slice_size; in Compute()