Searched refs:b_base_ptr (Results 1 – 3 of 3) sorted by relevance
354 auto* b_base_ptr = in_y.template flat<Scalar>().data();370 b_device_memory.push_back(AsDeviceMemory(b_base_ptr));378 b_device_memory.push_back(AsDeviceMemory(b_base_ptr + i * k * n));391 b_device_memory.push_back(AsDeviceMemory(b_base_ptr + i * k * n));526 auto* b_base_ptr = in_y.template flat<Scalar>().data();543 b_device_memory.push_back(AsDeviceMemory(b_base_ptr));551 b_device_memory.push_back(AsDeviceMemory(b_base_ptr + i * k * n));564 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);
552 const T* b_base_ptr = b_t.template flat<T>().data(); in Compute() local567 b_base_ptr = b_conj_t.template flat<T>().data(); in Compute()579 const T* b_i = b_base_ptr + i * b_slice_size; in Compute()