Searched refs:host_buf (Results 1 – 2 of 2) sorted by relevance
59 gtl::InlinedVector<int32, 24> host_buf(ndims * 3); in TileSimple()63 host_buf[i] = in_strides[i]; in TileSimple()64 host_buf[ndims + i] = out_strides[i]; in TileSimple()65 host_buf[ndims * 2 + i] = in.dim_size(i); in TileSimple()69 auto num_bytes = sizeof(int64) * host_buf.size(); in TileSimple()73 d.memcpyHostToDevice(dev_buf, host_buf.data(), num_bytes); in TileSimple()
64 gtl::InlinedVector<int32, 24> host_buf(ndims * 3); in TransposeSimple() local69 host_buf[i] = in_strides[i]; in TransposeSimple()70 host_buf[ndims + i] = out_strides[i]; in TransposeSimple()71 host_buf[ndims * 2 + i] = perm[i]; in TransposeSimple()74 auto num_bytes = sizeof(int64) * host_buf.size(); in TransposeSimple()78 d.memcpyHostToDevice(dev_buf, host_buf.data(), num_bytes); in TransposeSimple()