2017-02-23 06:26:44 +08:00
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
< html xmlns = "http://www.w3.org/1999/xhtml" >
< head >
< meta http-equiv = "Content-Type" content = "text/xhtml;charset=UTF-8" / >
< meta http-equiv = "X-UA-Compatible" content = "IE=9" / >
< meta name = "generator" content = "Doxygen 1.8.5" / >
2017-03-21 01:33:21 +08:00
< title > Faiss: /data/users/matthijs/github_faiss/faiss/gpu/utils/Tensor.cuh Source File< / title >
2017-02-23 06:26:44 +08:00
< link href = "tabs.css" rel = "stylesheet" type = "text/css" / >
< script type = "text/javascript" src = "jquery.js" > < / script >
< script type = "text/javascript" src = "dynsections.js" > < / script >
< link href = "search/search.css" rel = "stylesheet" type = "text/css" / >
< script type = "text/javascript" src = "search/search.js" > < / script >
< script type = "text/javascript" >
$(document).ready(function() { searchBox.OnSelectItem(0); });
< / script >
< link href = "doxygen.css" rel = "stylesheet" type = "text/css" / >
< / head >
< body >
< div id = "top" > <!-- do not remove this div, it is closed by doxygen! -->
< div id = "titlearea" >
< table cellspacing = "0" cellpadding = "0" >
< tbody >
< tr style = "height: 56px;" >
< td style = "padding-left: 0.5em;" >
< div id = "projectname" > Faiss
< / div >
< / td >
< / tr >
< / tbody >
< / table >
< / div >
<!-- end header part -->
<!-- Generated by Doxygen 1.8.5 -->
< script type = "text/javascript" >
var searchBox = new SearchBox("searchBox", "search",false,'Search');
< / script >
< div id = "navrow1" class = "tabs" >
< ul class = "tablist" >
< li > < a href = "index.html" > < span > Main  Page< / span > < / a > < / li >
< li > < a href = "namespaces.html" > < span > Namespaces< / span > < / a > < / li >
< li > < a href = "annotated.html" > < span > Classes< / span > < / a > < / li >
< li class = "current" > < a href = "files.html" > < span > Files< / span > < / a > < / li >
< li >
< div id = "MSearchBox" class = "MSearchBoxInactive" >
< span class = "left" >
< img id = "MSearchSelect" src = "search/mag_sel.png"
onmouseover="return searchBox.OnSearchSelectShow()"
onmouseout="return searchBox.OnSearchSelectHide()"
alt=""/>
< input type = "text" id = "MSearchField" value = "Search" accesskey = "S"
onfocus="searchBox.OnSearchFieldFocus(true)"
onblur="searchBox.OnSearchFieldFocus(false)"
onkeyup="searchBox.OnSearchFieldChange(event)"/>
< / span > < span class = "right" >
< a id = "MSearchClose" href = "javascript:searchBox.CloseResultsWindow()" > < img id = "MSearchCloseImg" border = "0" src = "search/close.png" alt = "" / > < / a >
< / span >
< / div >
< / li >
< / ul >
< / div >
< div id = "navrow2" class = "tabs2" >
< ul class = "tablist" >
< li > < a href = "files.html" > < span > File  List< / span > < / a > < / li >
< / ul >
< / div >
<!-- window showing the filter options -->
< div id = "MSearchSelectWindow"
onmouseover="return searchBox.OnSearchSelectShow()"
onmouseout="return searchBox.OnSearchSelectHide()"
onkeydown="return searchBox.OnSearchSelectKey(event)">
< a class = "SelectItem" href = "javascript:void(0)" onclick = "searchBox.OnSelectItem(0)" > < span class = "SelectionMark" >   < / span > All< / a > < a class = "SelectItem" href = "javascript:void(0)" onclick = "searchBox.OnSelectItem(1)" > < span class = "SelectionMark" >   < / span > Classes< / a > < a class = "SelectItem" href = "javascript:void(0)" onclick = "searchBox.OnSelectItem(2)" > < span class = "SelectionMark" >   < / span > Namespaces< / a > < a class = "SelectItem" href = "javascript:void(0)" onclick = "searchBox.OnSelectItem(3)" > < span class = "SelectionMark" >   < / span > Functions< / a > < a class = "SelectItem" href = "javascript:void(0)" onclick = "searchBox.OnSelectItem(4)" > < span class = "SelectionMark" >   < / span > Variables< / a > < a class = "SelectItem" href = "javascript:void(0)" onclick = "searchBox.OnSelectItem(5)" > < span class = "SelectionMark" >   < / span > Typedefs< / a > < a class = "SelectItem" href = "javascript:void(0)" onclick = "searchBox.OnSelectItem(6)" > < span class = "SelectionMark" >   < / span > Enumerations< / a > < a class = "SelectItem" href = "javascript:void(0)" onclick = "searchBox.OnSelectItem(7)" > < span class = "SelectionMark" >   < / span > Enumerator< / a > < a class = "SelectItem" href = "javascript:void(0)" onclick = "searchBox.OnSelectItem(8)" > < span class = "SelectionMark" >   < / span > Friends< / a > < / div >
<!-- iframe showing the search results (closed by default) -->
< div id = "MSearchResultsWindow" >
< iframe src = "javascript:void(0)" frameborder = "0"
name="MSearchResults" id="MSearchResults">
< / iframe >
< / div >
< div id = "nav-path" class = "navpath" >
< ul >
< li class = "navelem" > < a class = "el" href = "dir_6b3ae6988449b0834e9596fad5d75199.html" > gpu< / a > < / li > < li class = "navelem" > < a class = "el" href = "dir_498271007b03b2a0521055e88776887b.html" > utils< / a > < / li > < / ul >
< / div >
< / div > <!-- top -->
< div class = "header" >
< div class = "headertitle" >
< div class = "title" > Tensor.cuh< / div > < / div >
< / div > <!-- header -->
< div class = "contents" >
< div class = "fragment" > < div class = "line" > < a name = "l00001" > < / a > < span class = "lineno" > 1< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00002" > < / a > < span class = "lineno" > 2< / span >   < span class = "comment" > /**< / span > < / div >
< div class = "line" > < a name = "l00003" > < / a > < span class = "lineno" > 3< / span >   < span class = "comment" > * Copyright (c) 2015-present, Facebook, Inc.< / span > < / div >
< div class = "line" > < a name = "l00004" > < / a > < span class = "lineno" > 4< / span >   < span class = "comment" > * All rights reserved.< / span > < / div >
< div class = "line" > < a name = "l00005" > < / a > < span class = "lineno" > 5< / span >   < span class = "comment" > *< / span > < / div >
< div class = "line" > < a name = "l00006" > < / a > < span class = "lineno" > 6< / span >   < span class = "comment" > * This source code is licensed under the CC-by-NC license found in the< / span > < / div >
< div class = "line" > < a name = "l00007" > < / a > < span class = "lineno" > 7< / span >   < span class = "comment" > * LICENSE file in the root directory of this source tree.< / span > < / div >
< div class = "line" > < a name = "l00008" > < / a > < span class = "lineno" > 8< / span >   < span class = "comment" > */< / span > < / div >
< div class = "line" > < a name = "l00009" > < / a > < span class = "lineno" > 9< / span >   < / div >
< div class = "line" > < a name = "l00010" > < / a > < span class = "lineno" > 10< / span >   < span class = "comment" > // Copyright 2004-present Facebook. All Rights Reserved.< / span > < / div >
< div class = "line" > < a name = "l00011" > < / a > < span class = "lineno" > 11< / span >   < / div >
< div class = "line" > < a name = "l00012" > < / a > < span class = "lineno" > 12< / span >   < span class = "preprocessor" > #pragma once< / span > < / div >
< div class = "line" > < a name = "l00013" > < / a > < span class = "lineno" > 13< / span >   < span class = "preprocessor" > < / span > < / div >
< div class = "line" > < a name = "l00014" > < / a > < span class = "lineno" > 14< / span >   < span class = "preprocessor" > #include < assert.h> < / span > < / div >
< div class = "line" > < a name = "l00015" > < / a > < span class = "lineno" > 15< / span >   < span class = "preprocessor" > #include < cuda.h> < / span > < / div >
< div class = "line" > < a name = "l00016" > < / a > < span class = "lineno" > 16< / span >   < span class = "preprocessor" > #include < cuda_runtime.h> < / span > < / div >
< div class = "line" > < a name = "l00017" > < / a > < span class = "lineno" > 17< / span >   < span class = "preprocessor" > #include < initializer_list> < / span > < / div >
< div class = "line" > < a name = "l00018" > < / a > < span class = "lineno" > 18< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00019" > < / a > < span class = "lineno" > 19< / span >   < span class = "comment" > /// Multi-dimensional array class for CUDA device and host usage.< / span > < / div >
< div class = "line" > < a name = "l00020" > < / a > < span class = "lineno" > 20< / span >   < span class = "comment" > /// Originally from Facebook' s fbcunn, since added to the Torch GPU< / span > < / div >
< div class = "line" > < a name = "l00021" > < / a > < span class = "lineno" > 21< / span >   < span class = "comment" > /// library cutorch as well.< / span > < / div >
< div class = "line" > < a name = "l00022" > < / a > < span class = "lineno" > 22< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00023" > < / a > < span class = "lineno" > 23< / span >   < span class = "keyword" > namespace < / span > faiss { < span class = "keyword" > namespace < / span > gpu {< / div >
< div class = "line" > < a name = "l00024" > < / a > < span class = "lineno" > 24< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00025" > < / a > < span class = "lineno" > 25< / span >   < span class = "comment" > /// Our tensor type< / span > < / div >
< div class = "line" > < a name = "l00026" > < / a > < span class = "lineno" > 26< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T,< / div >
< div class = "line" > < a name = "l00027" > < / a > < span class = "lineno" > 27< / span >   < span class = "keywordtype" > int< / span > Dim,< / div >
< div class = "line" > < a name = "l00028" > < / a > < span class = "lineno" > 28< / span >   < span class = "keywordtype" > bool< / span > Contig,< / div >
< div class = "line" > < a name = "l00029" > < / a > < span class = "lineno" > 29< / span >   < span class = "keyword" > typename< / span > IndexT,< / div >
< div class = "line" > < a name = "l00030" > < / a > < span class = "lineno" > 30< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < span class = "keyword" > class < / span > PtrTraits> < / div >
< div class = "line" > < a name = "l00031" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html" > 31< / a > < / span >   < span class = "keyword" > class < / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< / a > ;< / div >
< div class = "line" > < a name = "l00032" > < / a > < span class = "lineno" > 32< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00033" > < / a > < span class = "lineno" > 33< / span >   < span class = "comment" > /// Type of a subspace of a tensor< / span > < / div >
< div class = "line" > < a name = "l00034" > < / a > < span class = "lineno" > < a class = "line" href = "namespacefaiss_1_1gpu_1_1detail.html" > 34< / a > < / span >   < span class = "comment" > < / span > < span class = "keyword" > namespace < / span > detail {< / div >
< div class = "line" > < a name = "l00035" > < / a > < span class = "lineno" > 35< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > TensorType,< / div >
< div class = "line" > < a name = "l00036" > < / a > < span class = "lineno" > 36< / span >   < span class = "keywordtype" > int< / span > SubDim,< / div >
< div class = "line" > < a name = "l00037" > < / a > < span class = "lineno" > 37< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < span class = "keyword" > class < / span > PtrTraits> < / div >
< div class = "line" > < a name = "l00038" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > 38< / a > < / span >   < span class = "keyword" > class < / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > ;< / div >
< div class = "line" > < a name = "l00039" > < / a > < span class = "lineno" > 39< / span >   }< / div >
< div class = "line" > < a name = "l00040" > < / a > < span class = "lineno" > 40< / span >   < / div >
< div class = "line" > < a name = "l00041" > < / a > < span class = "lineno" > 41< / span >   < span class = "keyword" > namespace < / span > traits {< / div >
< div class = "line" > < a name = "l00042" > < / a > < span class = "lineno" > 42< / span >   < / div >
< div class = "line" > < a name = "l00043" > < / a > < span class = "lineno" > 43< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00044" > < / a > < span class = "lineno" > < a class = "line" href = "structfaiss_1_1gpu_1_1traits_1_1RestrictPtrTraits.html" > 44< / a > < / span >   < span class = "keyword" > struct < / span > < a class = "code" href = "structfaiss_1_1gpu_1_1traits_1_1RestrictPtrTraits.html" > RestrictPtrTraits< / a > {< / div >
< div class = "line" > < a name = "l00045" > < / a > < span class = "lineno" > 45< / span >   < span class = "keyword" > typedef< / span > T* __restrict__ PtrType;< / div >
< div class = "line" > < a name = "l00046" > < / a > < span class = "lineno" > 46< / span >   };< / div >
< div class = "line" > < a name = "l00047" > < / a > < span class = "lineno" > 47< / span >   < / div >
< div class = "line" > < a name = "l00048" > < / a > < span class = "lineno" > 48< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00049" > < / a > < span class = "lineno" > < a class = "line" href = "structfaiss_1_1gpu_1_1traits_1_1DefaultPtrTraits.html" > 49< / a > < / span >   < span class = "keyword" > struct < / span > < a class = "code" href = "structfaiss_1_1gpu_1_1traits_1_1DefaultPtrTraits.html" > DefaultPtrTraits< / a > {< / div >
< div class = "line" > < a name = "l00050" > < / a > < span class = "lineno" > 50< / span >   < span class = "keyword" > typedef< / span > T* PtrType;< / div >
< div class = "line" > < a name = "l00051" > < / a > < span class = "lineno" > 51< / span >   };< / div >
< div class = "line" > < a name = "l00052" > < / a > < span class = "lineno" > 52< / span >   < / div >
< div class = "line" > < a name = "l00053" > < / a > < span class = "lineno" > 53< / span >   }< / div >
< div class = "line" > < a name = "l00054" > < / a > < span class = "lineno" > 54< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00055" > < / a > < span class = "lineno" > 55< / span >   < span class = "comment" > /**< / span > < / div >
< div class = "line" > < a name = "l00056" > < / a > < span class = "lineno" > 56< / span >   < span class = "comment" > Templated multi-dimensional array that supports strided access of< / span > < / div >
< div class = "line" > < a name = "l00057" > < / a > < span class = "lineno" > 57< / span >   < span class = "comment" > elements. Main access is through `operator[]`; e.g.,< / span > < / div >
< div class = "line" > < a name = "l00058" > < / a > < span class = "lineno" > 58< / span >   < span class = "comment" > `tensor[x][y][z]`.< / span > < / div >
< div class = "line" > < a name = "l00059" > < / a > < span class = "lineno" > 59< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00060" > < / a > < span class = "lineno" > 60< / span >   < span class = "comment" > - `T` is the contained type (e.g., `float`)< / span > < / div >
< div class = "line" > < a name = "l00061" > < / a > < span class = "lineno" > 61< / span >   < span class = "comment" > - `Dim` is the tensor rank< / span > < / div >
< div class = "line" > < a name = "l00062" > < / a > < span class = "lineno" > 62< / span >   < span class = "comment" > - If `Contig` is true, then the tensor is assumed to be< / span > < / div >
< div class = "line" > < a name = "l00063" > < / a > < span class = "lineno" > 63< / span >   < span class = "comment" > - contiguous, and only operations that make sense on contiguous< / span > < / div >
< div class = "line" > < a name = "l00064" > < / a > < span class = "lineno" > 64< / span >   < span class = "comment" > - arrays are allowed (e.g., no transpose). Strides are still< / span > < / div >
< div class = "line" > < a name = "l00065" > < / a > < span class = "lineno" > 65< / span >   < span class = "comment" > - calculated, but innermost stride is assumed to be 1.< / span > < / div >
< div class = "line" > < a name = "l00066" > < / a > < span class = "lineno" > 66< / span >   < span class = "comment" > - `IndexT` is the integer type used for size/stride arrays, and for< / span > < / div >
< div class = "line" > < a name = "l00067" > < / a > < span class = "lineno" > 67< / span >   < span class = "comment" > - all indexing math. Default is `int`, but for large tensors, `long`< / span > < / div >
< div class = "line" > < a name = "l00068" > < / a > < span class = "lineno" > 68< / span >   < span class = "comment" > - can be used instead.< / span > < / div >
< div class = "line" > < a name = "l00069" > < / a > < span class = "lineno" > 69< / span >   < span class = "comment" > - `PtrTraits` are traits applied to our data pointer (T*). By default,< / span > < / div >
< div class = "line" > < a name = "l00070" > < / a > < span class = "lineno" > 70< / span >   < span class = "comment" > - this is just T*, but RestrictPtrTraits can be used to apply T*< / span > < / div >
< div class = "line" > < a name = "l00071" > < / a > < span class = "lineno" > 71< / span >   < span class = "comment" > - __restrict__ for alias-free analysis.< / span > < / div >
< div class = "line" > < a name = "l00072" > < / a > < span class = "lineno" > 72< / span >   < span class = "comment" > */< / span > < / div >
< div class = "line" > < a name = "l00073" > < / a > < span class = "lineno" > 73< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T,< / div >
< div class = "line" > < a name = "l00074" > < / a > < span class = "lineno" > 74< / span >   < span class = "keywordtype" > int< / span > Dim,< / div >
< div class = "line" > < a name = "l00075" > < / a > < span class = "lineno" > 75< / span >   < span class = "keywordtype" > bool< / span > Contig = < span class = "keyword" > false< / span > ,< / div >
< div class = "line" > < a name = "l00076" > < / a > < span class = "lineno" > 76< / span >   < span class = "keyword" > typename< / span > IndexT = int,< / div >
< div class = "line" > < a name = "l00077" > < / a > < span class = "lineno" > 77< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < span class = "keyword" > class < / span > PtrTraits = < a class = "code" href = "structfaiss_1_1gpu_1_1traits_1_1DefaultPtrTraits.html" > traits::DefaultPtrTraits< / a > > < / div >
< div class = "line" > < a name = "l00078" > < / a > < span class = "lineno" > 78< / span >   < span class = "keyword" > class < / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< / a > {< / div >
< div class = "line" > < a name = "l00079" > < / a > < span class = "lineno" > 79< / span >   < span class = "keyword" > public< / span > :< / div >
< div class = "line" > < a name = "l00080" > < / a > < span class = "lineno" > 80< / span >   < span class = "keyword" > enum< / span > { NumDim = Dim };< / div >
< div class = "line" > < a name = "l00081" > < / a > < span class = "lineno" > 81< / span >   < span class = "keyword" > typedef< / span > T DataType;< / div >
< div class = "line" > < a name = "l00082" > < / a > < span class = "lineno" > 82< / span >   < span class = "keyword" > typedef< / span > IndexT IndexType;< / div >
< div class = "line" > < a name = "l00083" > < / a > < span class = "lineno" > 83< / span >   < span class = "keyword" > enum< / span > { IsContig = Contig };< / div >
< div class = "line" > < a name = "l00084" > < / a > < span class = "lineno" > 84< / span >   < span class = "keyword" > typedef< / span > < span class = "keyword" > typename< / span > PtrTraits< T> ::PtrType DataPtrType;< / div >
< div class = "line" > < a name = "l00085" > < / a > < span class = "lineno" > 85< / span >   < span class = "keyword" > typedef< / span > Tensor< T, Dim, Contig, IndexT, PtrTraits> TensorType;< / div >
< div class = "line" > < a name = "l00086" > < / a > < span class = "lineno" > 86< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00087" > < / a > < span class = "lineno" > 87< / span >   < span class = "comment" > /// Default constructor< / span > < / div >
< div class = "line" > < a name = "l00088" > < / a > < span class = "lineno" > 88< / span >   < span class = "comment" > < / span > __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a151fd78dd5695b1346739e32a9e76309" > Tensor< / a > ();< / div >
< div class = "line" > < a name = "l00089" > < / a > < span class = "lineno" > 89< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00090" > < / a > < span class = "lineno" > 90< / span >   < span class = "comment" > /// Copy constructor< / span > < / div >
< div class = "line" > < a name = "l00091" > < / a > < span class = "lineno" > 91< / span >   < span class = "comment" > < / span > __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a151fd78dd5695b1346739e32a9e76309" > Tensor< / a > (Tensor< T, Dim, Contig, IndexT, PtrTraits> & t)< / div >
< div class = "line" > < a name = "l00092" > < / a > < span class = "lineno" > 92< / span >   = < span class = "keywordflow" > default< / span > ;< / div >
< div class = "line" > < a name = "l00093" > < / a > < span class = "lineno" > 93< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00094" > < / a > < span class = "lineno" > 94< / span >   < span class = "comment" > /// Move constructor< / span > < / div >
< div class = "line" > < a name = "l00095" > < / a > < span class = "lineno" > 95< / span >   < span class = "comment" > < / span > __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a151fd78dd5695b1346739e32a9e76309" > Tensor< / a > (Tensor< T, Dim, Contig, IndexT, PtrTraits> & & t)< / div >
< div class = "line" > < a name = "l00096" > < / a > < span class = "lineno" > 96< / span >   = < span class = "keywordflow" > default< / span > ;< / div >
< div class = "line" > < a name = "l00097" > < / a > < span class = "lineno" > 97< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00098" > < / a > < span class = "lineno" > 98< / span >   < span class = "comment" > /// Assignment< / span > < / div >
< div class = "line" > < a name = "l00099" > < / a > < span class = "lineno" > 99< / span >   < span class = "comment" > < / span > __host__ __device__ Tensor< T, Dim, Contig, IndexT, PtrTraits> & < / div >
< div class = "line" > < a name = "l00100" > < / a > < span class = "lineno" > 100< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a55896db55a03e43a1789886aa22a0ba4" > operator=< / a > (Tensor< T, Dim, Contig, IndexT, PtrTraits> & t) = < span class = "keywordflow" > default< / span > ;< / div >
< div class = "line" > < a name = "l00101" > < / a > < span class = "lineno" > 101< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00102" > < / a > < span class = "lineno" > 102< / span >   < span class = "comment" > /// Move assignment< / span > < / div >
< div class = "line" > < a name = "l00103" > < / a > < span class = "lineno" > 103< / span >   < span class = "comment" > < / span > __host__ __device__ Tensor< T, Dim, Contig, IndexT, PtrTraits> & < / div >
< div class = "line" > < a name = "l00104" > < / a > < span class = "lineno" > 104< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a55896db55a03e43a1789886aa22a0ba4" > operator=< / a > (Tensor< T, Dim, Contig, IndexT, PtrTraits> & & t);< / div >
< div class = "line" > < a name = "l00105" > < / a > < span class = "lineno" > 105< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00106" > < / a > < span class = "lineno" > 106< / span >   < span class = "comment" > /// Constructor that calculates strides with no padding< / span > < / div >
< div class = "line" > < a name = "l00107" > < / a > < span class = "lineno" > 107< / span >   < span class = "comment" > < / span > __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a151fd78dd5695b1346739e32a9e76309" > Tensor< / a > (DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a37d3ac0dffcaad29f09d6e85fb07b335" > data< / a > ,< / div >
< div class = "line" > < a name = "l00108" > < / a > < span class = "lineno" > 108< / span >   < span class = "keyword" > const< / span > IndexT < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#ac3e0306267da475d8140bf6d12cf4af6" > sizes< / a > [Dim]);< / div >
< div class = "line" > < a name = "l00109" > < / a > < span class = "lineno" > 109< / span >   __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a151fd78dd5695b1346739e32a9e76309" > Tensor< / a > (DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a37d3ac0dffcaad29f09d6e85fb07b335" > data< / a > ,< / div >
< div class = "line" > < a name = "l00110" > < / a > < span class = "lineno" > 110< / span >   std::initializer_list< IndexT> < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#ac3e0306267da475d8140bf6d12cf4af6" > sizes< / a > );< / div >
< div class = "line" > < a name = "l00111" > < / a > < span class = "lineno" > 111< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00112" > < / a > < span class = "lineno" > 112< / span >   < span class = "comment" > /// Constructor that takes arbitrary size/stride arrays.< / span > < / div >
< div class = "line" > < a name = "l00113" > < / a > < span class = "lineno" > 113< / span >   < span class = "comment" > /// Errors if you attempt to pass non-contiguous strides to a< / span > < / div >
< div class = "line" > < a name = "l00114" > < / a > < span class = "lineno" > 114< / span >   < span class = "comment" > /// contiguous tensor.< / span > < / div >
< div class = "line" > < a name = "l00115" > < / a > < span class = "lineno" > 115< / span >   < span class = "comment" > < / span > __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a151fd78dd5695b1346739e32a9e76309" > Tensor< / a > (DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a37d3ac0dffcaad29f09d6e85fb07b335" > data< / a > ,< / div >
< div class = "line" > < a name = "l00116" > < / a > < span class = "lineno" > 116< / span >   < span class = "keyword" > const< / span > IndexT < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#ac3e0306267da475d8140bf6d12cf4af6" > sizes< / a > [Dim],< / div >
< div class = "line" > < a name = "l00117" > < / a > < span class = "lineno" > 117< / span >   < span class = "keyword" > const< / span > IndexT < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a9a0a293f287e9cba68f31c9e57b79c64" > strides< / a > [Dim]);< / div >
< div class = "line" > < a name = "l00118" > < / a > < span class = "lineno" > 118< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00119" > < / a > < span class = "lineno" > 119< / span >   < span class = "comment" > /// Copies a tensor into ourselves; sizes must match< / span > < / div >
< div class = "line" > < a name = "l00120" > < / a > < span class = "lineno" > 120< / span >   < span class = "comment" > < / span > __host__ < span class = "keywordtype" > void< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#ae981a94263044f38be89d690dd958426" > copyFrom< / a > (Tensor< T, Dim, Contig, IndexT, PtrTraits> & t,< / div >
< div class = "line" > < a name = "l00121" > < / a > < span class = "lineno" > 121< / span >   cudaStream_t stream);< / div >
< div class = "line" > < a name = "l00122" > < / a > < span class = "lineno" > 122< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00123" > < / a > < span class = "lineno" > 123< / span >   < span class = "comment" > /// Copies ourselves into a tensor; sizes must match< / span > < / div >
< div class = "line" > < a name = "l00124" > < / a > < span class = "lineno" > 124< / span >   < span class = "comment" > < / span > __host__ < span class = "keywordtype" > void< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a6ce60d784817469d5b47b560da177b92" > copyTo< / a > (Tensor< T, Dim, Contig, IndexT, PtrTraits> & t,< / div >
< div class = "line" > < a name = "l00125" > < / a > < span class = "lineno" > 125< / span >   cudaStream_t stream);< / div >
< div class = "line" > < a name = "l00126" > < / a > < span class = "lineno" > 126< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00127" > < / a > < span class = "lineno" > 127< / span >   < span class = "comment" > /// Returns true if the two tensors are of the same dimensionality,< / span > < / div >
< div class = "line" > < a name = "l00128" > < / a > < span class = "lineno" > 128< / span >   < span class = "comment" > /// size and stride.< / span > < / div >
< div class = "line" > < a name = "l00129" > < / a > < span class = "lineno" > 129< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keywordtype" > int< / span > OtherDim> < / div >
< div class = "line" > < a name = "l00130" > < / a > < span class = "lineno" > 130< / span >   __host__ __device__ < span class = "keywordtype" > bool< / span > < / div >
< div class = "line" > < a name = "l00131" > < / a > < span class = "lineno" > 131< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a013fac5bd3c78e90b35a6f46dbf3ffda" > isSame< / a > (< span class = "keyword" > const< / span > Tensor< T, OtherDim, Contig, IndexT, PtrTraits> & rhs) < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00132" > < / a > < span class = "lineno" > 132< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00133" > < / a > < span class = "lineno" > 133< / span >   < span class = "comment" > /// Cast to a tensor of a different type of the same size and< / span > < / div >
< div class = "line" > < a name = "l00134" > < / a > < span class = "lineno" > 134< / span >   < span class = "comment" > /// stride. U and our type T must be of the same size< / span > < / div >
< div class = "line" > < a name = "l00135" > < / a > < span class = "lineno" > 135< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < / div >
< div class = "line" > < a name = "l00136" > < / a > < span class = "lineno" > 136< / span >   __host__ __device__ Tensor< U, Dim, Contig, IndexT, PtrTraits> < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#ab462574aa0756b701e52279c13449764" > cast< / a > ();< / div >
< div class = "line" > < a name = "l00137" > < / a > < span class = "lineno" > 137< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00138" > < / a > < span class = "lineno" > 138< / span >   < span class = "comment" > /// Const version of `cast`< / span > < / div >
< div class = "line" > < a name = "l00139" > < / a > < span class = "lineno" > 139< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < / div >
< div class = "line" > < a name = "l00140" > < / a > < span class = "lineno" > 140< / span >   __host__ __device__< / div >
< div class = "line" > < a name = "l00141" > < / a > < span class = "lineno" > 141< / span >   < span class = "keyword" > const< / span > Tensor< U, Dim, Contig, IndexT, PtrTraits> < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#ab462574aa0756b701e52279c13449764" > cast< / a > () < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00142" > < / a > < span class = "lineno" > 142< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00143" > < / a > < span class = "lineno" > 143< / span >   < span class = "comment" > /// Cast to a tensor of a different type which is potentially a< / span > < / div >
< div class = "line" > < a name = "l00144" > < / a > < span class = "lineno" > 144< / span >   < span class = "comment" > /// different size than our type T. Tensor must be aligned and the< / span > < / div >
< div class = "line" > < a name = "l00145" > < / a > < span class = "lineno" > 145< / span >   < span class = "comment" > /// innermost dimension must be a size that is a multiple of< / span > < / div >
< div class = "line" > < a name = "l00146" > < / a > < span class = "lineno" > 146< / span >   < span class = "comment" > /// sizeof(U) / sizeof(T), and the stride of the innermost dimension< / span > < / div >
< div class = "line" > < a name = "l00147" > < / a > < span class = "lineno" > 147< / span >   < span class = "comment" > /// must be contiguous. The stride of all outer dimensions must be a< / span > < / div >
< div class = "line" > < a name = "l00148" > < / a > < span class = "lineno" > 148< / span >   < span class = "comment" > /// multiple of sizeof(U) / sizeof(T) as well.< / span > < / div >
< div class = "line" > < a name = "l00149" > < / a > < span class = "lineno" > 149< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < / div >
< div class = "line" > < a name = "l00150" > < / a > < span class = "lineno" > 150< / span >   __host__ __device__ Tensor< U, Dim, Contig, IndexT, PtrTraits> < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#aab60b0711646ffe587b62a5434d53c6a" > castResize< / a > ();< / div >
< div class = "line" > < a name = "l00151" > < / a > < span class = "lineno" > 151< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00152" > < / a > < span class = "lineno" > 152< / span >   < span class = "comment" > /// Const version of `castResize`< / span > < / div >
< div class = "line" > < a name = "l00153" > < / a > < span class = "lineno" > 153< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < / div >
< div class = "line" > < a name = "l00154" > < / a > < span class = "lineno" > 154< / span >   __host__ __device__ < span class = "keyword" > const< / span > Tensor< U, Dim, Contig, IndexT, PtrTraits> < / div >
< div class = "line" > < a name = "l00155" > < / a > < span class = "lineno" > 155< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#aab60b0711646ffe587b62a5434d53c6a" > castResize< / a > () < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00156" > < / a > < span class = "lineno" > 156< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00157" > < / a > < span class = "lineno" > 157< / span >   < span class = "comment" > /// Returns true if we can castResize() this tensor to the new type< / span > < / div >
< div class = "line" > < a name = "l00158" > < / a > < span class = "lineno" > 158< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < / div >
< div class = "line" > < a name = "l00159" > < / a > < span class = "lineno" > 159< / span >   __host__ __device__ < span class = "keywordtype" > bool< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a9b3a5fb3a903295b7d2f8cb2caae0352" > canCastResize< / a > () < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00160" > < / a > < span class = "lineno" > 160< / span >   < span class = "comment" > < / span > < / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00161" > < / a > < span class = "lineno" > 161< / span >   < span class = "comment" > /// Attempts to cast this tensor to a tensor of a different IndexT.< / span > < / div >
< div class = "line" > < a name = "l00162" > < / a > < span class = "lineno" > 162< / span >   < span class = "comment" > /// Fails if size or stride entries are not representable in the new< / span > < / div >
< div class = "line" > < a name = "l00163" > < / a > < span class = "lineno" > 163< / span >   < span class = "comment" > /// IndexT.< / span > < / div >
< div class = "line" > < a name = "l00164" > < / a > < span class = "lineno" > 164< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > NewIndexT> < / div >
< div class = "line" > < a name = "l00165" > < / a > < span class = "lineno" > 165< / span >   __host__ Tensor< T, Dim, Contig, NewIndexT, PtrTraits> < / div >
< div class = "line" > < a name = "l00166" > < / a > < span class = "lineno" > 166< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#afa66c4592bc873a3f663d765cb9ea2c0" > castIndexType< / a > () < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00167" > < / a > < span class = "lineno" > 167< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00168" > < / a > < span class = "lineno" > 168< / span >   < span class = "comment" > /// Returns true if we can castIndexType() this tensor to the new< / span > < / div >
< div class = "line" > < a name = "l00169" > < / a > < span class = "lineno" > 169< / span >   < span class = "comment" > /// index type< / span > < / div >
< div class = "line" > < a name = "l00170" > < / a > < span class = "lineno" > 170< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > NewIndexT> < / div >
< div class = "line" > < a name = "l00171" > < / a > < span class = "lineno" > 171< / span >   __host__ < span class = "keywordtype" > bool< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a080cf59b73bee015c02947009e70ce07" > canCastIndexType< / a > () < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00172" > < / a > < span class = "lineno" > 172< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00173" > < / a > < span class = "lineno" > 173< / span >   < span class = "comment" > /// Returns a raw pointer to the start of our data.< / span > < / div >
< div class = "line" > < a name = "l00174" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#a37d3ac0dffcaad29f09d6e85fb07b335" > 174< / a > < / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a37d3ac0dffcaad29f09d6e85fb07b335" > data< / a > () {< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00175" > < / a > < span class = "lineno" > 175< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a08024dbcaed3c1235cac62b476f14509" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00176" > < / a > < span class = "lineno" > 176< / span >   }< / div >
< div class = "line" > < a name = "l00177" > < / a > < span class = "lineno" > 177< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00178" > < / a > < span class = "lineno" > 178< / span >   < span class = "comment" > /// Returns a raw pointer to the end of our data, assuming< / span > < / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00179" > < / a > < span class = "lineno" > 179< / span >   < span class = "comment" > /// continuity< / span > < / div >
< div class = "line" > < a name = "l00180" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#ad35e0c816162fd08e372b21b79cab6c1" > 180< / a > < / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#ad35e0c816162fd08e372b21b79cab6c1" > end< / a > () {< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00181" > < / a > < span class = "lineno" > 181< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a37d3ac0dffcaad29f09d6e85fb07b335" > data< / a > () + < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a41f7f08246bf8584929693bb1896205f" > numElements< / a > ();< / div >
< div class = "line" > < a name = "l00182" > < / a > < span class = "lineno" > 182< / span >   }< / div >
< div class = "line" > < a name = "l00183" > < / a > < span class = "lineno" > 183< / span >   < span class = "comment" > < / span > < / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00184" > < / a > < span class = "lineno" > 184< / span >   < span class = "comment" > /// Returns a raw pointer to the start of our data (const).< / span > < / div >
< div class = "line" > < a name = "l00185" > < / a > < span class = "lineno" > 185< / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00186" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#a9a098dad7e211328d1ae8b00d02839ce" > 186< / a > < / span >   < span class = "keyword" > const< / span > DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a9a098dad7e211328d1ae8b00d02839ce" > data< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00187" > < / a > < span class = "lineno" > 187< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a08024dbcaed3c1235cac62b476f14509" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00188" > < / a > < span class = "lineno" > 188< / span >   }< / div >
< div class = "line" > < a name = "l00189" > < / a > < span class = "lineno" > 189< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00190" > < / a > < span class = "lineno" > 190< / span >   < span class = "comment" > /// Returns a raw pointer to the end of our data, assuming< / span > < / div >
< div class = "line" > < a name = "l00191" > < / a > < span class = "lineno" > 191< / span >   < span class = "comment" > /// continuity (const)< / span > < / div >
< div class = "line" > < a name = "l00192" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#a00d2d17504f63b9e834afac730a64324" > 192< / a > < / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a00d2d17504f63b9e834afac730a64324" > end< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00193" > < / a > < span class = "lineno" > 193< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a37d3ac0dffcaad29f09d6e85fb07b335" > data< / a > () + < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a41f7f08246bf8584929693bb1896205f" > numElements< / a > ();< / div >
< div class = "line" > < a name = "l00194" > < / a > < span class = "lineno" > 194< / span >   }< / div >
< div class = "line" > < a name = "l00195" > < / a > < span class = "lineno" > 195< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00196" > < / a > < span class = "lineno" > 196< / span >   < span class = "comment" > /// Cast to a different datatype< / span > < / div >
< div class = "line" > < a name = "l00197" > < / a > < span class = "lineno" > 197< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < / div >
< div class = "line" > < a name = "l00198" > < / a > < span class = "lineno" > 198< / span >   __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00199" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#a7c987316708113dafd11971092d59f69" > 199< / a > < / span >   < span class = "keyword" > typename< / span > PtrTraits< U> ::PtrType < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a7c987316708113dafd11971092d59f69" > dataAs< / a > () {< / div >
< div class = "line" > < a name = "l00200" > < / a > < span class = "lineno" > 200< / span >   < span class = "keywordflow" > return< / span > < span class = "keyword" > reinterpret_cast< < / span > typename PtrTraits< U> ::PtrType< span class = "keyword" > > < / span > (< a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a08024dbcaed3c1235cac62b476f14509" > data_< / a > );< / div >
< div class = "line" > < a name = "l00201" > < / a > < span class = "lineno" > 201< / span >   }< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00202" > < / a > < span class = "lineno" > 202< / span >   < span class = "comment" > < / span > < / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00203" > < / a > < span class = "lineno" > 203< / span >   < span class = "comment" > /// Cast to a different datatype< / span > < / div >
< div class = "line" > < a name = "l00204" > < / a > < span class = "lineno" > 204< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < / div >
< div class = "line" > < a name = "l00205" > < / a > < span class = "lineno" > 205< / span >   __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00206" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#ad02cd32977ebf89636c8d085b4470c2e" > 206< / a > < / span >   < span class = "keyword" > const< / span > < span class = "keyword" > typename< / span > PtrTraits< const U> ::PtrType < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#ad02cd32977ebf89636c8d085b4470c2e" > dataAs< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00207" > < / a > < span class = "lineno" > 207< / span >   < span class = "keywordflow" > return< / span > < span class = "keyword" > reinterpret_cast< < / span > typename PtrTraits< const U> ::PtrType< span class = "keyword" > > < / span > (< a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a08024dbcaed3c1235cac62b476f14509" > data_< / a > );< / div >
< div class = "line" > < a name = "l00208" > < / a > < span class = "lineno" > 208< / span >   }< / div >
< div class = "line" > < a name = "l00209" > < / a > < span class = "lineno" > 209< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00210" > < / a > < span class = "lineno" > 210< / span >   < span class = "comment" > /// Returns a read/write view of a portion of our tensor.< / span > < / div >
< div class = "line" > < a name = "l00211" > < / a > < span class = "lineno" > 211< / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00212" > < / a > < span class = "lineno" > 212< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > detail::SubTensor< / a > < TensorType, Dim - 1, PtrTraits> < / div >
< div class = "line" > < a name = "l00213" > < / a > < span class = "lineno" > 213< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a204ea1c3232b51512a9c027bb44ee3c4" > operator[]< / a > (IndexT);< / div >
< div class = "line" > < a name = "l00214" > < / a > < span class = "lineno" > 214< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00215" > < / a > < span class = "lineno" > 215< / span >   < span class = "comment" > /// Returns a read/write view of a portion of our tensor (const).< / span > < / div >
< div class = "line" > < a name = "l00216" > < / a > < span class = "lineno" > 216< / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00217" > < / a > < span class = "lineno" > 217< / span >   < span class = "keyword" > const< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > detail::SubTensor< / a > < TensorType, Dim - 1, PtrTraits> < / div >
< div class = "line" > < a name = "l00218" > < / a > < span class = "lineno" > 218< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a204ea1c3232b51512a9c027bb44ee3c4" > operator[]< / a > (IndexT) < span class = "keyword" > const< / span > ;< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00219" > < / a > < span class = "lineno" > 219< / span >   < span class = "comment" > < / span > < / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00220" > < / a > < span class = "lineno" > 220< / span >   < span class = "comment" > /// Returns the size of a given dimension, `[0, Dim - 1]`. No bounds< / span > < / div >
< div class = "line" > < a name = "l00221" > < / a > < span class = "lineno" > 221< / span >   < span class = "comment" > /// checking.< / span > < / div >
< div class = "line" > < a name = "l00222" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#a2d276c97faf432cdc9f3552da63c0d3c" > 222< / a > < / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > IndexT < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a2d276c97faf432cdc9f3552da63c0d3c" > getSize< / a > (< span class = "keywordtype" > int< / span > i)< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00223" > < / a > < span class = "lineno" > 223< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#afe1bd60ecb861eabafaad7362a68ec1f" > size_< / a > [i];< / div >
< div class = "line" > < a name = "l00224" > < / a > < span class = "lineno" > 224< / span >   }< / div >
< div class = "line" > < a name = "l00225" > < / a > < span class = "lineno" > 225< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00226" > < / a > < span class = "lineno" > 226< / span >   < span class = "comment" > /// Returns the stride of a given dimension, `[0, Dim - 1]`. No bounds< / span > < / div >
< div class = "line" > < a name = "l00227" > < / a > < span class = "lineno" > 227< / span >   < span class = "comment" > /// checking.< / span > < / div >
< div class = "line" > < a name = "l00228" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#a57855c6e0c529a309168833017049ab5" > 228< / a > < / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > IndexT < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a57855c6e0c529a309168833017049ab5" > getStride< / a > (< span class = "keywordtype" > int< / span > i)< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00229" > < / a > < span class = "lineno" > 229< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a46c20e6be3ce16feb99f8d3a5ee268c9" > stride_< / a > [i];< / div >
< div class = "line" > < a name = "l00230" > < / a > < span class = "lineno" > 230< / span >   }< / div >
< div class = "line" > < a name = "l00231" > < / a > < span class = "lineno" > 231< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00232" > < / a > < span class = "lineno" > 232< / span >   < span class = "comment" > /// Returns the total number of elements contained within our data< / span > < / div >
< div class = "line" > < a name = "l00233" > < / a > < span class = "lineno" > 233< / span >   < span class = "comment" > /// (product of `getSize(i)`)< / span > < / div >
< div class = "line" > < a name = "l00234" > < / a > < span class = "lineno" > 234< / span >   < span class = "comment" > < / span > __host__ __device__ IndexT < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a41f7f08246bf8584929693bb1896205f" > numElements< / a > () < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00235" > < / a > < span class = "lineno" > 235< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00236" > < / a > < span class = "lineno" > 236< / span >   < span class = "comment" > /// If we are contiguous, returns the total size in bytes of our< / span > < / div >
< div class = "line" > < a name = "l00237" > < / a > < span class = "lineno" > 237< / span >   < span class = "comment" > /// data< / span > < / div >
< div class = "line" > < a name = "l00238" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#aae42c9c020682fd094958976d8e610ad" > 238< / a > < / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keywordtype" > size_t< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#aae42c9c020682fd094958976d8e610ad" > getSizeInBytes< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00239" > < / a > < span class = "lineno" > 239< / span >   < span class = "keywordflow" > return< / span > (< span class = "keywordtype" > size_t< / span > ) < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a41f7f08246bf8584929693bb1896205f" > numElements< / a > () * < span class = "keyword" > sizeof< / span > (T);< / div >
< div class = "line" > < a name = "l00240" > < / a > < span class = "lineno" > 240< / span >   }< / div >
< div class = "line" > < a name = "l00241" > < / a > < span class = "lineno" > 241< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00242" > < / a > < span class = "lineno" > 242< / span >   < span class = "comment" > /// Returns the size array.< / span > < / div >
< div class = "line" > < a name = "l00243" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#ac3e0306267da475d8140bf6d12cf4af6" > 243< / a > < / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > < span class = "keyword" > const< / span > IndexT* < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#ac3e0306267da475d8140bf6d12cf4af6" > sizes< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00244" > < / a > < span class = "lineno" > 244< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#afe1bd60ecb861eabafaad7362a68ec1f" > size_< / a > ;< / div >
< div class = "line" > < a name = "l00245" > < / a > < span class = "lineno" > 245< / span >   }< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00246" > < / a > < span class = "lineno" > 246< / span >   < span class = "comment" > < / span > < / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00247" > < / a > < span class = "lineno" > 247< / span >   < span class = "comment" > /// Returns the stride array.< / span > < / div >
< div class = "line" > < a name = "l00248" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#a9a0a293f287e9cba68f31c9e57b79c64" > 248< / a > < / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > < span class = "keyword" > const< / span > IndexT* < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a9a0a293f287e9cba68f31c9e57b79c64" > strides< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00249" > < / a > < span class = "lineno" > 249< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a46c20e6be3ce16feb99f8d3a5ee268c9" > stride_< / a > ;< / div >
< div class = "line" > < a name = "l00250" > < / a > < span class = "lineno" > 250< / span >   }< / div >
< div class = "line" > < a name = "l00251" > < / a > < span class = "lineno" > 251< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00252" > < / a > < span class = "lineno" > 252< / span >   < span class = "comment" > /// Returns true if there is no padding within the tensor and no< / span > < / div >
< div class = "line" > < a name = "l00253" > < / a > < span class = "lineno" > 253< / span >   < span class = "comment" > /// re-ordering of the dimensions.< / span > < / div >
< div class = "line" > < a name = "l00254" > < / a > < span class = "lineno" > 254< / span >   < span class = "comment" > /// ~~~< / span > < / div >
< div class = "line" > < a name = "l00255" > < / a > < span class = "lineno" > 255< / span >   < span class = "comment" > /// (stride(i) == size(i + 1) * stride(i + 1)) & & stride(dim - 1) == 0< / span > < / div >
< div class = "line" > < a name = "l00256" > < / a > < span class = "lineno" > 256< / span >   < span class = "comment" > /// ~~~< / span > < / div >
< div class = "line" > < a name = "l00257" > < / a > < span class = "lineno" > 257< / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keywordtype" > bool< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#acd445331bcc0373cf925f7a02660107c" > isContiguous< / a > () < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00258" > < / a > < span class = "lineno" > 258< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00259" > < / a > < span class = "lineno" > 259< / span >   < span class = "comment" > /// Returns whether a given dimension has only increasing stride< / span > < / div >
< div class = "line" > < a name = "l00260" > < / a > < span class = "lineno" > 260< / span >   < span class = "comment" > /// from the previous dimension. A tensor that was permuted by< / span > < / div >
< div class = "line" > < a name = "l00261" > < / a > < span class = "lineno" > 261< / span >   < span class = "comment" > /// exchanging size and stride only will fail this check.< / span > < / div >
< div class = "line" > < a name = "l00262" > < / a > < span class = "lineno" > 262< / span >   < span class = "comment" > /// If `i == 0` just check `size > 0`. Returns `false` if `stride` is `< = 0`.< / span > < / div >
< div class = "line" > < a name = "l00263" > < / a > < span class = "lineno" > 263< / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keywordtype" > bool< / span > isConsistentlySized(< span class = "keywordtype" > int< / span > i) < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00264" > < / a > < span class = "lineno" > 264< / span >   < / div >
< div class = "line" > < a name = "l00265" > < / a > < span class = "lineno" > 265< / span >   < span class = "comment" > // Returns whether at each dimension `stride < = size`.< / span > < / div >
< div class = "line" > < a name = "l00266" > < / a > < span class = "lineno" > 266< / span >   < span class = "comment" > // If this is not the case then iterating once over the size space will< / span > < / div >
< div class = "line" > < a name = "l00267" > < / a > < span class = "lineno" > 267< / span >   < span class = "comment" > // touch the same memory locations multiple times.< / span > < / div >
< div class = "line" > < a name = "l00268" > < / a > < span class = "lineno" > 268< / span >   __host__ __device__ < span class = "keywordtype" > bool< / span > isConsistentlySized() < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00269" > < / a > < span class = "lineno" > 269< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00270" > < / a > < span class = "lineno" > 270< / span >   < span class = "comment" > /// Returns true if the given dimension index has no padding< / span > < / div >
< div class = "line" > < a name = "l00271" > < / a > < span class = "lineno" > 271< / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keywordtype" > bool< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a31075cd6df497e4c1a7c7e47fa2603ca" > isContiguousDim< / a > (< span class = "keywordtype" > int< / span > i) < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00272" > < / a > < span class = "lineno" > 272< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00273" > < / a > < span class = "lineno" > 273< / span >   < span class = "comment" > /// Returns a tensor of the same dimension after transposing the two< / span > < / div >
< div class = "line" > < a name = "l00274" > < / a > < span class = "lineno" > 274< / span >   < span class = "comment" > /// dimensions given. Does not actually move elements; transposition< / span > < / div >
< div class = "line" > < a name = "l00275" > < / a > < span class = "lineno" > 275< / span >   < span class = "comment" > /// is made by permuting the size/stride arrays.< / span > < / div >
< div class = "line" > < a name = "l00276" > < / a > < span class = "lineno" > 276< / span >   < span class = "comment" > /// If the dimensions are not valid, asserts.< / span > < / div >
< div class = "line" > < a name = "l00277" > < / a > < span class = "lineno" > 277< / span >   < span class = "comment" > < / span > __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< T, Dim, Contig, IndexT, PtrTraits> < / a > < / div >
< div class = "line" > < a name = "l00278" > < / a > < span class = "lineno" > 278< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#aa8b48391a291a7e214c7204d81cd9d73" > transpose< / a > (< span class = "keywordtype" > int< / span > dim1, < span class = "keywordtype" > int< / span > dim2) < span class = "keyword" > const< / span > ;< / div >
< div class = "line" > < a name = "l00279" > < / a > < span class = "lineno" > 279< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00280" > < / a > < span class = "lineno" > 280< / span >   < span class = "comment" > /// Upcast a tensor of dimension `D` to some tensor of dimension< / span > < / div >
< div class = "line" > < a name = "l00281" > < / a > < span class = "lineno" > 281< / span >   < span class = "comment" > /// D' > D by padding the leading dimensions by 1< / span > < / div >
< div class = "line" > < a name = "l00282" > < / a > < span class = "lineno" > 282< / span >   < span class = "comment" > /// e.g., upcasting a 2-d tensor `[2][3]` to a 4-d tensor `[1][1][2][3]`< / span > < / div >
< div class = "line" > < a name = "l00283" > < / a > < span class = "lineno" > 283< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keywordtype" > int< / span > NewDim> < / div >
< div class = "line" > < a name = "l00284" > < / a > < span class = "lineno" > 284< / span >   __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< T, NewDim, Contig, IndexT, PtrTraits> < / a > < / div >
< div class = "line" > < a name = "l00285" > < / a > < span class = "lineno" > 285< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#ad6df1805b76f74018bc94fffab8987ac" > upcastOuter< / a > ();< / div >
< div class = "line" > < a name = "l00286" > < / a > < span class = "lineno" > 286< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00287" > < / a > < span class = "lineno" > 287< / span >   < span class = "comment" > /// Upcast a tensor of dimension `D` to some tensor of dimension< / span > < / div >
< div class = "line" > < a name = "l00288" > < / a > < span class = "lineno" > 288< / span >   < span class = "comment" > /// D' > D by padding the lowest/most varying dimensions by 1< / span > < / div >
< div class = "line" > < a name = "l00289" > < / a > < span class = "lineno" > 289< / span >   < span class = "comment" > /// e.g., upcasting a 2-d tensor `[2][3]` to a 4-d tensor `[2][3][1][1]`< / span > < / div >
< div class = "line" > < a name = "l00290" > < / a > < span class = "lineno" > 290< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keywordtype" > int< / span > NewDim> < / div >
< div class = "line" > < a name = "l00291" > < / a > < span class = "lineno" > 291< / span >   __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< T, NewDim, Contig, IndexT, PtrTraits> < / a > < / div >
< div class = "line" > < a name = "l00292" > < / a > < span class = "lineno" > 292< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a70308137b566362d4cc37d3e71b3ae98" > upcastInner< / a > ();< / div >
< div class = "line" > < a name = "l00293" > < / a > < span class = "lineno" > 293< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00294" > < / a > < span class = "lineno" > 294< / span >   < span class = "comment" > /// Downcast a tensor of dimension `D` to some tensor of dimension< / span > < / div >
< div class = "line" > < a name = "l00295" > < / a > < span class = "lineno" > 295< / span >   < span class = "comment" > /// D' < D by collapsing the leading dimensions. asserts if there is< / span > < / div >
< div class = "line" > < a name = "l00296" > < / a > < span class = "lineno" > 296< / span >   < span class = "comment" > /// padding on the leading dimensions.< / span > < / div >
< div class = "line" > < a name = "l00297" > < / a > < span class = "lineno" > 297< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keywordtype" > int< / span > NewDim> < / div >
< div class = "line" > < a name = "l00298" > < / a > < span class = "lineno" > 298< / span >   __host__ __device__< / div >
< div class = "line" > < a name = "l00299" > < / a > < span class = "lineno" > 299< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< T, NewDim, Contig, IndexT, PtrTraits> < / a > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a5591c571ac52482fc7ceea6ef99af794" > downcastOuter< / a > ();< / div >
< div class = "line" > < a name = "l00300" > < / a > < span class = "lineno" > 300< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00301" > < / a > < span class = "lineno" > 301< / span >   < span class = "comment" > /// Downcast a tensor of dimension `D` to some tensor of dimension< / span > < / div >
< div class = "line" > < a name = "l00302" > < / a > < span class = "lineno" > 302< / span >   < span class = "comment" > /// D' < D by collapsing the leading dimensions. asserts if there is< / span > < / div >
< div class = "line" > < a name = "l00303" > < / a > < span class = "lineno" > 303< / span >   < span class = "comment" > /// padding on the leading dimensions.< / span > < / div >
< div class = "line" > < a name = "l00304" > < / a > < span class = "lineno" > 304< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keywordtype" > int< / span > NewDim> < / div >
< div class = "line" > < a name = "l00305" > < / a > < span class = "lineno" > 305< / span >   __host__ __device__< / div >
< div class = "line" > < a name = "l00306" > < / a > < span class = "lineno" > 306< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< T, NewDim, Contig, IndexT, PtrTraits> < / a > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a4bcfb3f04077a42e0d8c924e11b2d637" > downcastInner< / a > ();< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00307" > < / a > < span class = "lineno" > 307< / span >   < span class = "comment" > < / span > < / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00308" > < / a > < span class = "lineno" > 308< / span >   < span class = "comment" > /// Returns a tensor that is a view of the `SubDim`-dimensional slice< / span > < / div >
< div class = "line" > < a name = "l00309" > < / a > < span class = "lineno" > 309< / span >   < span class = "comment" > /// of this tensor, starting at `at`.< / span > < / div >
< div class = "line" > < a name = "l00310" > < / a > < span class = "lineno" > 310< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keywordtype" > int< / span > SubDim> < / div >
< div class = "line" > < a name = "l00311" > < / a > < span class = "lineno" > 311< / span >   __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< T, SubDim, Contig, IndexT, PtrTraits> < / a > < / div >
< div class = "line" > < a name = "l00312" > < / a > < span class = "lineno" > 312< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a67ccf916f78dacf0ae0d4816a663eb3f" > view< / a > (DataPtrType at);< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00313" > < / a > < span class = "lineno" > 313< / span >   < span class = "comment" > < / span > < / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00314" > < / a > < span class = "lineno" > 314< / span >   < span class = "comment" > /// Returns a tensor that is a view of the `SubDim`-dimensional slice< / span > < / div >
< div class = "line" > < a name = "l00315" > < / a > < span class = "lineno" > 315< / span >   < span class = "comment" > /// of this tensor, starting where our data begins< / span > < / div >
< div class = "line" > < a name = "l00316" > < / a > < span class = "lineno" > 316< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keywordtype" > int< / span > SubDim> < / div >
< div class = "line" > < a name = "l00317" > < / a > < span class = "lineno" > 317< / span >   __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< T, SubDim, Contig, IndexT, PtrTraits> < / a > < / div >
< div class = "line" > < a name = "l00318" > < / a > < span class = "lineno" > 318< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a67ccf916f78dacf0ae0d4816a663eb3f" > view< / a > ();< / div >
< div class = "line" > < a name = "l00319" > < / a > < span class = "lineno" > 319< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00320" > < / a > < span class = "lineno" > 320< / span >   < span class = "comment" > /// Returns a tensor of the same dimension that is a view of the< / span > < / div >
< div class = "line" > < a name = "l00321" > < / a > < span class = "lineno" > 321< / span >   < span class = "comment" > /// original tensor with the specified dimension restricted to the< / span > < / div >
< div class = "line" > < a name = "l00322" > < / a > < span class = "lineno" > 322< / span >   < span class = "comment" > /// elements in the range [start, start + size)< / span > < / div >
< div class = "line" > < a name = "l00323" > < / a > < span class = "lineno" > 323< / span >   < span class = "comment" > < / span > __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< T, Dim, Contig, IndexT, PtrTraits> < / a > < / div >
< div class = "line" > < a name = "l00324" > < / a > < span class = "lineno" > 324< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a782cd978394660371a22229f02e49e60" > narrowOutermost< / a > (IndexT start, IndexT size);< / div >
< div class = "line" > < a name = "l00325" > < / a > < span class = "lineno" > 325< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00326" > < / a > < span class = "lineno" > 326< / span >   < span class = "comment" > /// Returns a tensor of the same dimension that is a view of the< / span > < / div >
< div class = "line" > < a name = "l00327" > < / a > < span class = "lineno" > 327< / span >   < span class = "comment" > /// original tensor with the specified dimension restricted to the< / span > < / div >
< div class = "line" > < a name = "l00328" > < / a > < span class = "lineno" > 328< / span >   < span class = "comment" > /// elements in the range [start, start + size).< / span > < / div >
< div class = "line" > < a name = "l00329" > < / a > < span class = "lineno" > 329< / span >   < span class = "comment" > /// Can occur in an arbitrary dimension, and is possibly< / span > < / div >
< div class = "line" > < a name = "l00330" > < / a > < span class = "lineno" > 330< / span >   < span class = "comment" > /// non-contiguous< / span > < / div >
< div class = "line" > < a name = "l00331" > < / a > < span class = "lineno" > 331< / span >   < span class = "comment" > < / span > __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< T, Dim, false, IndexT, PtrTraits> < / a > < / div >
< div class = "line" > < a name = "l00332" > < / a > < span class = "lineno" > 332< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a58d7a953431def329bac11374679c2dd" > narrow< / a > (< span class = "keywordtype" > int< / span > dim, IndexT start, IndexT size);< / div >
< div class = "line" > < a name = "l00333" > < / a > < span class = "lineno" > 333< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00334" > < / a > < span class = "lineno" > 334< / span >   < span class = "comment" > /// Returns a view of the given tensor expressed as a tensor of a< / span > < / div >
< div class = "line" > < a name = "l00335" > < / a > < span class = "lineno" > 335< / span >   < span class = "comment" > /// different number of dimensions.< / span > < / div >
< div class = "line" > < a name = "l00336" > < / a > < span class = "lineno" > 336< / span >   < span class = "comment" > /// Only works if we are contiguous.< / span > < / div >
< div class = "line" > < a name = "l00337" > < / a > < span class = "lineno" > 337< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keywordtype" > int< / span > NewDim> < / div >
< div class = "line" > < a name = "l00338" > < / a > < span class = "lineno" > 338< / span >   __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< T, NewDim, Contig, IndexT, PtrTraits> < / a > < / div >
< div class = "line" > < a name = "l00339" > < / a > < span class = "lineno" > 339< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a67ccf916f78dacf0ae0d4816a663eb3f" > view< / a > (std::initializer_list< IndexT> < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#ac3e0306267da475d8140bf6d12cf4af6" > sizes< / a > );< / div >
< div class = "line" > < a name = "l00340" > < / a > < span class = "lineno" > 340< / span >   < / div >
< div class = "line" > < a name = "l00341" > < / a > < span class = "lineno" > 341< / span >   < span class = "keyword" > protected< / span > :< span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00342" > < / a > < span class = "lineno" > 342< / span >   < span class = "comment" > /// Raw pointer to where the tensor data begins< / span > < / div >
< div class = "line" > < a name = "l00343" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#a08024dbcaed3c1235cac62b476f14509" > 343< / a > < / span >   < span class = "comment" > < / span > DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a08024dbcaed3c1235cac62b476f14509" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00344" > < / a > < span class = "lineno" > 344< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00345" > < / a > < span class = "lineno" > 345< / span >   < span class = "comment" > /// Array of strides (in sizeof(T) terms) per each dimension< / span > < / div >
< div class = "line" > < a name = "l00346" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#a46c20e6be3ce16feb99f8d3a5ee268c9" > 346< / a > < / span >   < span class = "comment" > < / span > IndexT < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a46c20e6be3ce16feb99f8d3a5ee268c9" > stride_< / a > [Dim];< / div >
< div class = "line" > < a name = "l00347" > < / a > < span class = "lineno" > 347< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00348" > < / a > < span class = "lineno" > 348< / span >   < span class = "comment" > /// Size per each dimension< / span > < / div >
< div class = "line" > < a name = "l00349" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#afe1bd60ecb861eabafaad7362a68ec1f" > 349< / a > < / span >   < span class = "comment" > < / span > IndexT < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#afe1bd60ecb861eabafaad7362a68ec1f" > size_< / a > [Dim];< / div >
< div class = "line" > < a name = "l00350" > < / a > < span class = "lineno" > 350< / span >   };< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00351" > < / a > < span class = "lineno" > 351< / span >   < / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00352" > < / a > < span class = "lineno" > 352< / span >   < span class = "comment" > // Utilities for checking a collection of tensors< / span > < / div >
< div class = "line" > < a name = "l00353" > < / a > < span class = "lineno" > 353< / span >   < span class = "keyword" > namespace < / span > detail {< / div >
< div class = "line" > < a name = "l00354" > < / a > < span class = "lineno" > 354< / span >   < / div >
< div class = "line" > < a name = "l00355" > < / a > < span class = "lineno" > 355< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > IndexType> < / div >
< div class = "line" > < a name = "l00356" > < / a > < span class = "lineno" > 356< / span >   < span class = "keywordtype" > bool< / span > canCastIndexType() {< / div >
< div class = "line" > < a name = "l00357" > < / a > < span class = "lineno" > 357< / span >   < span class = "keywordflow" > return< / span > < span class = "keyword" > true< / span > ;< / div >
< div class = "line" > < a name = "l00358" > < / a > < span class = "lineno" > 358< / span >   }< / div >
< div class = "line" > < a name = "l00359" > < / a > < span class = "lineno" > 359< / span >   < / div >
< div class = "line" > < a name = "l00360" > < / a > < span class = "lineno" > 360< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > IndexType, < span class = "keyword" > typename< / span > T, < span class = "keyword" > typename< / span > ... U> < / div >
< div class = "line" > < a name = "l00361" > < / a > < span class = "lineno" > 361< / span >   < span class = "keywordtype" > bool< / span > canCastIndexType(< span class = "keyword" > const< / span > T& arg, < span class = "keyword" > const< / span > U& ... args) {< / div >
< div class = "line" > < a name = "l00362" > < / a > < span class = "lineno" > 362< / span >   < span class = "keywordflow" > return< / span > arg.canCastIndexType< IndexType> () & & < / div >
< div class = "line" > < a name = "l00363" > < / a > < span class = "lineno" > 363< / span >   canCastIndexType(args...);< / div >
< div class = "line" > < a name = "l00364" > < / a > < span class = "lineno" > 364< / span >   }< / div >
< div class = "line" > < a name = "l00365" > < / a > < span class = "lineno" > 365< / span >   < / div >
< div class = "line" > < a name = "l00366" > < / a > < span class = "lineno" > 366< / span >   } < span class = "comment" > // namespace detail< / span > < / div >
< div class = "line" > < a name = "l00367" > < / a > < span class = "lineno" > 367< / span >   < / div >
< div class = "line" > < a name = "l00368" > < / a > < span class = "lineno" > 368< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > IndexType, < span class = "keyword" > typename< / span > ... T> < / div >
< div class = "line" > < a name = "l00369" > < / a > < span class = "lineno" > 369< / span >   < span class = "keywordtype" > bool< / span > canCastIndexType(< span class = "keyword" > const< / span > T& ... args) {< / div >
< div class = "line" > < a name = "l00370" > < / a > < span class = "lineno" > 370< / span >   < span class = "keywordflow" > return< / span > detail::canCastIndexType(args...);< / div >
< div class = "line" > < a name = "l00371" > < / a > < span class = "lineno" > 371< / span >   }< / div >
< div class = "line" > < a name = "l00372" > < / a > < span class = "lineno" > 372< / span >   < / div >
< div class = "line" > < a name = "l00373" > < / a > < span class = "lineno" > 373< / span >   < span class = "keyword" > namespace < / span > detail {< / div >
< div class = "line" > < a name = "l00374" > < / a > < span class = "lineno" > 374< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00375" > < / a > < span class = "lineno" > 375< / span >   < span class = "comment" > /// Specialization for a view of a single value (0-dimensional)< / span > < / div >
< div class = "line" > < a name = "l00376" > < / a > < span class = "lineno" > 376< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > TensorType, < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < span class = "keyword" > class < / span > PtrTraits> < / div >
< div class = "line" > < a name = "l00377" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html" > 377< / a > < / span >   < span class = "keyword" > class < / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > < TensorType, 0, PtrTraits> {< / div >
< div class = "line" > < a name = "l00378" > < / a > < span class = "lineno" > 378< / span >   < span class = "keyword" > public< / span > :< / div >
< div class = "line" > < a name = "l00379" > < / a > < span class = "lineno" > 379< / span >   __host__ __device__ < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html" > SubTensor< TensorType, 0, PtrTraits> < / a > < / div >
< div class = "line" > < a name = "l00380" > < / a > < span class = "lineno" > 380< / span >   operator=(< span class = "keyword" > typename< / span > TensorType::DataType val) {< / div >
< div class = "line" > < a name = "l00381" > < / a > < span class = "lineno" > 381< / span >   *< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > = val;< / div >
< div class = "line" > < a name = "l00382" > < / a > < span class = "lineno" > 382< / span >   < span class = "keywordflow" > return< / span > *< span class = "keyword" > this< / span > ;< / div >
< div class = "line" > < a name = "l00383" > < / a > < span class = "lineno" > 383< / span >   }< / div >
< div class = "line" > < a name = "l00384" > < / a > < span class = "lineno" > 384< / span >   < / div >
< div class = "line" > < a name = "l00385" > < / a > < span class = "lineno" > 385< / span >   < span class = "comment" > // operator T& < / span > < / div >
< div class = "line" > < a name = "l00386" > < / a > < span class = "lineno" > 386< / span >   __host__ __device__ < span class = "keyword" > operator< / span > < span class = "keyword" > typename< / span > TensorType::DataType& () {< / div >
< div class = "line" > < a name = "l00387" > < / a > < span class = "lineno" > 387< / span >   < span class = "keywordflow" > return< / span > *< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00388" > < / a > < span class = "lineno" > 388< / span >   }< / div >
< div class = "line" > < a name = "l00389" > < / a > < span class = "lineno" > 389< / span >   < / div >
< div class = "line" > < a name = "l00390" > < / a > < span class = "lineno" > 390< / span >   < span class = "comment" > // const operator T& returning const T& < / span > < / div >
< div class = "line" > < a name = "l00391" > < / a > < span class = "lineno" > 391< / span >   __host__ __device__ < span class = "keyword" > operator< / span > < span class = "keyword" > const< / span > < span class = "keyword" > typename< / span > TensorType::DataType& () < span class = "keyword" > const< / span > {< / div >
< div class = "line" > < a name = "l00392" > < / a > < span class = "lineno" > 392< / span >   < span class = "keywordflow" > return< / span > *< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00393" > < / a > < span class = "lineno" > 393< / span >   }< / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00394" > < / a > < span class = "lineno" > 394< / span >   < / div >
< div class = "line" > < a name = "l00395" > < / a > < span class = "lineno" > 395< / span >   < span class = "comment" > // operator& returning T*< / span > < / div >
< div class = "line" > < a name = "l00396" > < / a > < span class = "lineno" > 396< / span >   __host__ __device__ < span class = "keyword" > typename< / span > TensorType::DataType* operator& () {< / div >
< div class = "line" > < a name = "l00397" > < / a > < span class = "lineno" > 397< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00398" > < / a > < span class = "lineno" > 398< / span >   }< / div >
< div class = "line" > < a name = "l00399" > < / a > < span class = "lineno" > 399< / span >   < / div >
< div class = "line" > < a name = "l00400" > < / a > < span class = "lineno" > 400< / span >   < span class = "comment" > // const operator& returning const T*< / span > < / div >
< div class = "line" > < a name = "l00401" > < / a > < span class = "lineno" > 401< / span >   __host__ __device__ < span class = "keyword" > const< / span > < span class = "keyword" > typename< / span > TensorType::DataType* operator& ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00402" > < / a > < span class = "lineno" > 402< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00403" > < / a > < span class = "lineno" > 403< / span >   }< / div >
< div class = "line" > < a name = "l00404" > < / a > < span class = "lineno" > 404< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00405" > < / a > < span class = "lineno" > 405< / span >   < span class = "comment" > /// Returns a raw accessor to our slice.< / span > < / div >
< div class = "line" > < a name = "l00406" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#aae8c90b402493f5656f94701157a7417" > 406< / a > < / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > < span class = "keyword" > typename< / span > TensorType::DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#aae8c90b402493f5656f94701157a7417" > data< / a > () {< / div >
< div class = "line" > < a name = "l00407" > < / a > < span class = "lineno" > 407< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00408" > < / a > < span class = "lineno" > 408< / span >   }< / div >
< div class = "line" > < a name = "l00409" > < / a > < span class = "lineno" > 409< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00410" > < / a > < span class = "lineno" > 410< / span >   < span class = "comment" > /// Returns a raw accessor to our slice (const).< / span > < / div >
< div class = "line" > < a name = "l00411" > < / a > < span class = "lineno" > 411< / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00412" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a750047ff919799af43b4861b580c82e3" > 412< / a > < / span >   < span class = "keyword" > const< / span > < span class = "keyword" > typename< / span > TensorType::DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a750047ff919799af43b4861b580c82e3" > data< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00413" > < / a > < span class = "lineno" > 413< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00414" > < / a > < span class = "lineno" > 414< / span >   }< / div >
< div class = "line" > < a name = "l00415" > < / a > < span class = "lineno" > 415< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00416" > < / a > < span class = "lineno" > 416< / span >   < span class = "comment" > /// Cast to a different datatype.< / span > < / div >
< div class = "line" > < a name = "l00417" > < / a > < span class = "lineno" > 417< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00418" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a67bfa92466e03834b7f007cb9cdf8d50" > 418< / a > < / span >   __host__ __device__ T& < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a67bfa92466e03834b7f007cb9cdf8d50" > as< / a > () {< / div >
< div class = "line" > < a name = "l00419" > < / a > < span class = "lineno" > 419< / span >   < span class = "keywordflow" > return< / span > *dataAs< T> ();< / div >
< div class = "line" > < a name = "l00420" > < / a > < span class = "lineno" > 420< / span >   }< / div >
< div class = "line" > < a name = "l00421" > < / a > < span class = "lineno" > 421< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00422" > < / a > < span class = "lineno" > 422< / span >   < span class = "comment" > /// Cast to a different datatype (const).< / span > < / div >
< div class = "line" > < a name = "l00423" > < / a > < span class = "lineno" > 423< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00424" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#ad1d375e64756991dadeb5a1e63ed2cfd" > 424< / a > < / span >   __host__ __device__ < span class = "keyword" > const< / span > T& < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#ad1d375e64756991dadeb5a1e63ed2cfd" > as< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00425" > < / a > < span class = "lineno" > 425< / span >   < span class = "keywordflow" > return< / span > *dataAs< T> ();< / div >
< div class = "line" > < a name = "l00426" > < / a > < span class = "lineno" > 426< / span >   }< / div >
< div class = "line" > < a name = "l00427" > < / a > < span class = "lineno" > 427< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00428" > < / a > < span class = "lineno" > 428< / span >   < span class = "comment" > /// Cast to a different datatype< / span > < / div >
< div class = "line" > < a name = "l00429" > < / a > < span class = "lineno" > 429< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00430" > < / a > < span class = "lineno" > 430< / span >   __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00431" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a23e80555a443797d60ae16d605dacd23" > 431< / a > < / span >   < span class = "keyword" > typename< / span > PtrTraits< T> ::PtrType < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a23e80555a443797d60ae16d605dacd23" > dataAs< / a > () {< / div >
< div class = "line" > < a name = "l00432" > < / a > < span class = "lineno" > 432< / span >   < span class = "keywordflow" > return< / span > < span class = "keyword" > reinterpret_cast< < / span > typename PtrTraits< T> ::PtrType< span class = "keyword" > > < / span > (< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > );< / div >
< div class = "line" > < a name = "l00433" > < / a > < span class = "lineno" > 433< / span >   }< / div >
< div class = "line" > < a name = "l00434" > < / a > < span class = "lineno" > 434< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00435" > < / a > < span class = "lineno" > 435< / span >   < span class = "comment" > /// Cast to a different datatype (const)< / span > < / div >
< div class = "line" > < a name = "l00436" > < / a > < span class = "lineno" > 436< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00437" > < / a > < span class = "lineno" > 437< / span >   __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00438" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a0ec3a48f265de627490e7cdf540e9fc5" > 438< / a > < / span >   < span class = "keyword" > typename< / span > PtrTraits< const T> ::PtrType < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a0ec3a48f265de627490e7cdf540e9fc5" > dataAs< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00439" > < / a > < span class = "lineno" > 439< / span >   < span class = "keywordflow" > return< / span > < span class = "keyword" > reinterpret_cast< < / span > typename PtrTraits< const T> ::PtrType< span class = "keyword" > > < / span > (< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > );< / div >
< div class = "line" > < a name = "l00440" > < / a > < span class = "lineno" > 440< / span >   }< / div >
< div class = "line" > < a name = "l00441" > < / a > < span class = "lineno" > 441< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00442" > < / a > < span class = "lineno" > 442< / span >   < span class = "comment" > /// Use the texture cache for reads< / span > < / div >
< div class = "line" > < a name = "l00443" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#ac44400045b113c527d6ed59a910f885c" > 443< / a > < / span >   < span class = "comment" > < / span > __device__ < span class = "keyword" > inline< / span > < span class = "keyword" > typename< / span > TensorType::DataType < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#ac44400045b113c527d6ed59a910f885c" > ldg< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00444" > < / a > < span class = "lineno" > 444< / span >   < span class = "preprocessor" > #if __CUDA_ARCH__ > = 350< / span > < / div >
< div class = "line" > < a name = "l00445" > < / a > < span class = "lineno" > 445< / span >   < span class = "preprocessor" > < / span > < span class = "keywordflow" > return< / span > __ldg(< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > );< / div >
< div class = "line" > < a name = "l00446" > < / a > < span class = "lineno" > 446< / span >   < span class = "preprocessor" > #else< / span > < / div >
< div class = "line" > < a name = "l00447" > < / a > < span class = "lineno" > 447< / span >   < span class = "preprocessor" > < / span > < span class = "keywordflow" > return< / span > *< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00448" > < / a > < span class = "lineno" > 448< / span >   < span class = "preprocessor" > #endif< / span > < / div >
< div class = "line" > < a name = "l00449" > < / a > < span class = "lineno" > 449< / span >   < span class = "preprocessor" > < / span > }< / div >
< div class = "line" > < a name = "l00450" > < / a > < span class = "lineno" > 450< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00451" > < / a > < span class = "lineno" > 451< / span >   < span class = "comment" > /// Use the texture cache for reads; cast as a particular type< / span > < / div >
< div class = "line" > < a name = "l00452" > < / a > < span class = "lineno" > 452< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00453" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#aa56767066e40a4758e37b26e43449f1d" > 453< / a > < / span >   __device__ < span class = "keyword" > inline< / span > T < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#aa56767066e40a4758e37b26e43449f1d" > ldgAs< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00454" > < / a > < span class = "lineno" > 454< / span >   < span class = "preprocessor" > #if __CUDA_ARCH__ > = 350< / span > < / div >
< div class = "line" > < a name = "l00455" > < / a > < span class = "lineno" > 455< / span >   < span class = "preprocessor" > < / span > < span class = "keywordflow" > return< / span > __ldg(dataAs< T> ());< / div >
< div class = "line" > < a name = "l00456" > < / a > < span class = "lineno" > 456< / span >   < span class = "preprocessor" > #else< / span > < / div >
< div class = "line" > < a name = "l00457" > < / a > < span class = "lineno" > 457< / span >   < span class = "preprocessor" > < / span > < span class = "keywordflow" > return< / span > as< T> ();< / div >
< div class = "line" > < a name = "l00458" > < / a > < span class = "lineno" > 458< / span >   < span class = "preprocessor" > #endif< / span > < / div >
< div class = "line" > < a name = "l00459" > < / a > < span class = "lineno" > 459< / span >   < span class = "preprocessor" > < / span > }< / div >
< div class = "line" > < a name = "l00460" > < / a > < span class = "lineno" > 460< / span >   < / div >
< div class = "line" > < a name = "l00461" > < / a > < span class = "lineno" > 461< / span >   < span class = "keyword" > protected< / span > :< span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00462" > < / a > < span class = "lineno" > 462< / span >   < span class = "comment" > /// One dimension greater can create us< / span > < / div >
< div class = "line" > < a name = "l00463" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a5aa8367d35e6c281c29855d7cf24bd6d" > 463< / a > < / span >   < span class = "comment" > < / span > < span class = "keyword" > friend< / span > < span class = "keyword" > class < / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > < TensorType, 1, PtrTraits> ;< / div >
< div class = "line" > < a name = "l00464" > < / a > < span class = "lineno" > 464< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00465" > < / a > < span class = "lineno" > 465< / span >   < span class = "comment" > /// Our parent tensor can create us< / span > < / div >
< div class = "line" > < a name = "l00466" > < / a > < span class = "lineno" > 466< / span >   < span class = "comment" > < / span > < span class = "keyword" > friend< / span > < span class = "keyword" > class < / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< / a > < typename TensorType::DataType,< / div >
< div class = "line" > < a name = "l00467" > < / a > < span class = "lineno" > 467< / span >   1,< / div >
< div class = "line" > < a name = "l00468" > < / a > < span class = "lineno" > 468< / span >   TensorType::IsContig,< / div >
< div class = "line" > < a name = "l00469" > < / a > < span class = "lineno" > 469< / span >   typename TensorType::IndexType,< / div >
< div class = "line" > < a name = "l00470" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a9118dea3cbf4ba3184500f6d870d89bb" > 470< / a > < / span >   PtrTraits> ;< / div >
< div class = "line" > < a name = "l00471" > < / a > < span class = "lineno" > 471< / span >   < / div >
< div class = "line" > < a name = "l00472" > < / a > < span class = "lineno" > 472< / span >   __host__ __device__ inline < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > (< / div >
< div class = "line" > < a name = "l00473" > < / a > < span class = "lineno" > 473< / span >   TensorType& t,< / div >
< div class = "line" > < a name = "l00474" > < / a > < span class = "lineno" > 474< / span >   typename TensorType::DataPtrType data)< / div >
< div class = "line" > < a name = "l00475" > < / a > < span class = "lineno" > 475< / span >   : < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a2d4e62fb08c180dfe2bde8d47361d61f" > tensor_< / a > (t),< / div >
< div class = "line" > < a name = "l00476" > < / a > < span class = "lineno" > 476< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > (data) {< / div >
< div class = "line" > < a name = "l00477" > < / a > < span class = "lineno" > 477< / span >   }< / div >
< div class = "line" > < a name = "l00478" > < / a > < span class = "lineno" > 478< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00479" > < / a > < span class = "lineno" > 479< / span >   < span class = "comment" > /// The tensor we' re referencing< / span > < / div >
< div class = "line" > < a name = "l00480" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a6e0f585a739cd1474ec24f56609d6501" > 480< / a > < / span >   < span class = "comment" > < / span > TensorType& < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a6e0f585a739cd1474ec24f56609d6501" > tensor_< / a > ;< / div >
< div class = "line" > < a name = "l00481" > < / a > < span class = "lineno" > 481< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00482" > < / a > < span class = "lineno" > 482< / span >   < span class = "comment" > /// Where our value is located< / span > < / div >
< div class = "line" > < a name = "l00483" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a6a548d4edb57d072be52cd827f055d6d" > 483< / a > < / span >   < span class = "comment" > < / span > < span class = "keyword" > typename< / span > TensorType::DataPtrType < span class = "keyword" > const< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a6a548d4edb57d072be52cd827f055d6d" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00484" > < / a > < span class = "lineno" > 484< / span >   };< / div >
< div class = "line" > < a name = "l00485" > < / a > < span class = "lineno" > 485< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00486" > < / a > < span class = "lineno" > 486< / span >   < span class = "comment" > /// A `SubDim`-rank slice of a parent Tensor< / span > < / div >
< div class = "line" > < a name = "l00487" > < / a > < span class = "lineno" > 487< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > TensorType,< / div >
< div class = "line" > < a name = "l00488" > < / a > < span class = "lineno" > 488< / span >   < span class = "keywordtype" > int< / span > SubDim,< / div >
< div class = "line" > < a name = "l00489" > < / a > < span class = "lineno" > 489< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < span class = "keyword" > class < / span > PtrTraits> < / div >
< div class = "line" > < a name = "l00490" > < / a > < span class = "lineno" > 490< / span >   < span class = "keyword" > class < / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > {< / div >
< div class = "line" > < a name = "l00491" > < / a > < span class = "lineno" > 491< / span >   < span class = "keyword" > public< / span > :< span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00492" > < / a > < span class = "lineno" > 492< / span >   < span class = "comment" > /// Returns a view of the data located at our offset (the dimension< / span > < / div >
< div class = "line" > < a name = "l00493" > < / a > < span class = "lineno" > 493< / span >   < span class = "comment" > /// `SubDim` - 1 tensor).< / span > < / div >
< div class = "line" > < a name = "l00494" > < / a > < span class = "lineno" > 494< / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00495" > < / a > < span class = "lineno" > 495< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > < TensorType, SubDim - 1, PtrTraits> < / div >
< div class = "line" > < a name = "l00496" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#ac722ca465d06da122898a07ce38276e2" > 496< / a > < / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#ac722ca465d06da122898a07ce38276e2" > operator[]< / a > (< span class = "keyword" > typename< / span > TensorType::IndexType index) {< / div >
< div class = "line" > < a name = "l00497" > < / a > < span class = "lineno" > 497< / span >   < span class = "keywordflow" > if< / span > (TensorType::IsContig & & SubDim == 1) {< / div >
< div class = "line" > < a name = "l00498" > < / a > < span class = "lineno" > 498< / span >   < span class = "comment" > // Innermost dimension is stride 1 for contiguous arrays< / span > < / div >
< div class = "line" > < a name = "l00499" > < / a > < span class = "lineno" > 499< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > < TensorType, SubDim - 1, PtrTraits> (< / div >
< div class = "line" > < a name = "l00500" > < / a > < span class = "lineno" > 500< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a2d4e62fb08c180dfe2bde8d47361d61f" > tensor_< / a > , < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > + index);< / div >
< div class = "line" > < a name = "l00501" > < / a > < span class = "lineno" > 501< / span >   } < span class = "keywordflow" > else< / span > {< / div >
< div class = "line" > < a name = "l00502" > < / a > < span class = "lineno" > 502< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > < TensorType, SubDim - 1, PtrTraits> (< / div >
< div class = "line" > < a name = "l00503" > < / a > < span class = "lineno" > 503< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a2d4e62fb08c180dfe2bde8d47361d61f" > tensor_< / a > ,< / div >
< div class = "line" > < a name = "l00504" > < / a > < span class = "lineno" > 504< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > + index * tensor_.getStride(TensorType::NumDim - SubDim));< / div >
< div class = "line" > < a name = "l00505" > < / a > < span class = "lineno" > 505< / span >   }< / div >
< div class = "line" > < a name = "l00506" > < / a > < span class = "lineno" > 506< / span >   }< / div >
< div class = "line" > < a name = "l00507" > < / a > < span class = "lineno" > 507< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00508" > < / a > < span class = "lineno" > 508< / span >   < span class = "comment" > /// Returns a view of the data located at our offset (the dimension< / span > < / div >
< div class = "line" > < a name = "l00509" > < / a > < span class = "lineno" > 509< / span >   < span class = "comment" > /// `SubDim` - 1 tensor) (const).< / span > < / div >
< div class = "line" > < a name = "l00510" > < / a > < span class = "lineno" > 510< / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00511" > < / a > < span class = "lineno" > 511< / span >   < span class = "keyword" > const< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > < TensorType, SubDim - 1, PtrTraits> < / div >
< div class = "line" > < a name = "l00512" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a4c3006fcd82c301b11505620e3e96378" > 512< / a > < / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a4c3006fcd82c301b11505620e3e96378" > operator[]< / a > (< span class = "keyword" > typename< / span > TensorType::IndexType index)< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00513" > < / a > < span class = "lineno" > 513< / span >   < span class = "keywordflow" > if< / span > (TensorType::IsContig & & SubDim == 1) {< / div >
< div class = "line" > < a name = "l00514" > < / a > < span class = "lineno" > 514< / span >   < span class = "comment" > // Innermost dimension is stride 1 for contiguous arrays< / span > < / div >
< div class = "line" > < a name = "l00515" > < / a > < span class = "lineno" > 515< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > < TensorType, SubDim - 1, PtrTraits> (< / div >
< div class = "line" > < a name = "l00516" > < / a > < span class = "lineno" > 516< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a2d4e62fb08c180dfe2bde8d47361d61f" > tensor_< / a > , < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > + index);< / div >
< div class = "line" > < a name = "l00517" > < / a > < span class = "lineno" > 517< / span >   } < span class = "keywordflow" > else< / span > {< / div >
< div class = "line" > < a name = "l00518" > < / a > < span class = "lineno" > 518< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > < TensorType, SubDim - 1, PtrTraits> (< / div >
< div class = "line" > < a name = "l00519" > < / a > < span class = "lineno" > 519< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a2d4e62fb08c180dfe2bde8d47361d61f" > tensor_< / a > ,< / div >
< div class = "line" > < a name = "l00520" > < / a > < span class = "lineno" > 520< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > + index * tensor_.getStride(TensorType::NumDim - SubDim));< / div >
< div class = "line" > < a name = "l00521" > < / a > < span class = "lineno" > 521< / span >   }< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00522" > < / a > < span class = "lineno" > 522< / span >   }< / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00523" > < / a > < span class = "lineno" > 523< / span >   < / div >
< div class = "line" > < a name = "l00524" > < / a > < span class = "lineno" > 524< / span >   < span class = "comment" > // operator& returning T*< / span > < / div >
< div class = "line" > < a name = "l00525" > < / a > < span class = "lineno" > 525< / span >   __host__ __device__ < span class = "keyword" > typename< / span > TensorType::DataType* operator& () {< / div >
< div class = "line" > < a name = "l00526" > < / a > < span class = "lineno" > 526< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00527" > < / a > < span class = "lineno" > 527< / span >   }< / div >
< div class = "line" > < a name = "l00528" > < / a > < span class = "lineno" > 528< / span >   < / div >
< div class = "line" > < a name = "l00529" > < / a > < span class = "lineno" > 529< / span >   < span class = "comment" > // const operator& returning const T*< / span > < / div >
< div class = "line" > < a name = "l00530" > < / a > < span class = "lineno" > 530< / span >   __host__ __device__ < span class = "keyword" > const< / span > < span class = "keyword" > typename< / span > TensorType::DataType* operator& ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00531" > < / a > < span class = "lineno" > 531< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00532" > < / a > < span class = "lineno" > 532< / span >   }< / div >
< div class = "line" > < a name = "l00533" > < / a > < span class = "lineno" > 533< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00534" > < / a > < span class = "lineno" > 534< / span >   < span class = "comment" > /// Returns a raw accessor to our slice.< / span > < / div >
< div class = "line" > < a name = "l00535" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a30dff4e7bea94cd894e17f6bdd7a7eb1" > 535< / a > < / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > < span class = "keyword" > typename< / span > TensorType::DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a30dff4e7bea94cd894e17f6bdd7a7eb1" > data< / a > () {< / div >
< div class = "line" > < a name = "l00536" > < / a > < span class = "lineno" > 536< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00537" > < / a > < span class = "lineno" > 537< / span >   }< / div >
< div class = "line" > < a name = "l00538" > < / a > < span class = "lineno" > 538< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00539" > < / a > < span class = "lineno" > 539< / span >   < span class = "comment" > /// Returns a raw accessor to our slice (const).< / span > < / div >
< div class = "line" > < a name = "l00540" > < / a > < span class = "lineno" > 540< / span >   < span class = "comment" > < / span > __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00541" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a6e7097578ba17c10895ec0dafa385901" > 541< / a > < / span >   < span class = "keyword" > const< / span > < span class = "keyword" > typename< / span > TensorType::DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a6e7097578ba17c10895ec0dafa385901" > data< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00542" > < / a > < span class = "lineno" > 542< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00543" > < / a > < span class = "lineno" > 543< / span >   }< / div >
< div class = "line" > < a name = "l00544" > < / a > < span class = "lineno" > 544< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00545" > < / a > < span class = "lineno" > 545< / span >   < span class = "comment" > /// Cast to a different datatype.< / span > < / div >
< div class = "line" > < a name = "l00546" > < / a > < span class = "lineno" > 546< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00547" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a0d32586e8f6f22f5f90bca566d901d0b" > 547< / a > < / span >   __host__ __device__ T& < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a0d32586e8f6f22f5f90bca566d901d0b" > as< / a > () {< / div >
< div class = "line" > < a name = "l00548" > < / a > < span class = "lineno" > 548< / span >   < span class = "keywordflow" > return< / span > *dataAs< T> ();< / div >
< div class = "line" > < a name = "l00549" > < / a > < span class = "lineno" > 549< / span >   }< / div >
< div class = "line" > < a name = "l00550" > < / a > < span class = "lineno" > 550< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00551" > < / a > < span class = "lineno" > 551< / span >   < span class = "comment" > /// Cast to a different datatype (const).< / span > < / div >
< div class = "line" > < a name = "l00552" > < / a > < span class = "lineno" > 552< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00553" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aefdafcf236e5c49ad3bce1646797f8f2" > 553< / a > < / span >   __host__ __device__ < span class = "keyword" > const< / span > T& < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aefdafcf236e5c49ad3bce1646797f8f2" > as< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00554" > < / a > < span class = "lineno" > 554< / span >   < span class = "keywordflow" > return< / span > *dataAs< T> ();< / div >
< div class = "line" > < a name = "l00555" > < / a > < span class = "lineno" > 555< / span >   }< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00556" > < / a > < span class = "lineno" > 556< / span >   < span class = "comment" > < / span > < / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00557" > < / a > < span class = "lineno" > 557< / span >   < span class = "comment" > /// Cast to a different datatype< / span > < / div >
< div class = "line" > < a name = "l00558" > < / a > < span class = "lineno" > 558< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00559" > < / a > < span class = "lineno" > 559< / span >   __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00560" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a9825bed624c3abb6337a1ab7654d7db7" > 560< / a > < / span >   < span class = "keyword" > typename< / span > PtrTraits< T> ::PtrType < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a9825bed624c3abb6337a1ab7654d7db7" > dataAs< / a > () {< / div >
< div class = "line" > < a name = "l00561" > < / a > < span class = "lineno" > 561< / span >   < span class = "keywordflow" > return< / span > < span class = "keyword" > reinterpret_cast< < / span > typename PtrTraits< T> ::PtrType< span class = "keyword" > > < / span > (< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > );< / div >
< div class = "line" > < a name = "l00562" > < / a > < span class = "lineno" > 562< / span >   }< / div >
< div class = "line" > < a name = "l00563" > < / a > < span class = "lineno" > 563< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00564" > < / a > < span class = "lineno" > 564< / span >   < span class = "comment" > /// Cast to a different datatype (const)< / span > < / div >
< div class = "line" > < a name = "l00565" > < / a > < span class = "lineno" > 565< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00566" > < / a > < span class = "lineno" > 566< / span >   __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00567" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a485abbadb5b5de23e88822366857a78f" > 567< / a > < / span >   < span class = "keyword" > typename< / span > PtrTraits< const T> ::PtrType < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a485abbadb5b5de23e88822366857a78f" > dataAs< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00568" > < / a > < span class = "lineno" > 568< / span >   < span class = "keywordflow" > return< / span > < span class = "keyword" > reinterpret_cast< < / span > typename PtrTraits< const T> ::PtrType< span class = "keyword" > > < / span > (< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > );< / div >
< div class = "line" > < a name = "l00569" > < / a > < span class = "lineno" > 569< / span >   }< / div >
2017-02-23 06:26:44 +08:00
< div class = "line" > < a name = "l00570" > < / a > < span class = "lineno" > 570< / span >   < span class = "comment" > < / span > < / div >
2017-03-21 01:33:21 +08:00
< div class = "line" > < a name = "l00571" > < / a > < span class = "lineno" > 571< / span >   < span class = "comment" > /// Use the texture cache for reads< / span > < / div >
< div class = "line" > < a name = "l00572" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a760782118b71504348d073ca1c92843a" > 572< / a > < / span >   < span class = "comment" > < / span > __device__ < span class = "keyword" > inline< / span > < span class = "keyword" > typename< / span > TensorType::DataType < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a760782118b71504348d073ca1c92843a" > ldg< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00573" > < / a > < span class = "lineno" > 573< / span >   < span class = "preprocessor" > #if __CUDA_ARCH__ > = 350< / span > < / div >
< div class = "line" > < a name = "l00574" > < / a > < span class = "lineno" > 574< / span >   < span class = "preprocessor" > < / span > < span class = "keywordflow" > return< / span > __ldg(< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > );< / div >
< div class = "line" > < a name = "l00575" > < / a > < span class = "lineno" > 575< / span >   < span class = "preprocessor" > #else< / span > < / div >
< div class = "line" > < a name = "l00576" > < / a > < span class = "lineno" > 576< / span >   < span class = "preprocessor" > < / span > < span class = "keywordflow" > return< / span > *< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00577" > < / a > < span class = "lineno" > 577< / span >   < span class = "preprocessor" > #endif< / span > < / div >
< div class = "line" > < a name = "l00578" > < / a > < span class = "lineno" > 578< / span >   < span class = "preprocessor" > < / span > }< / div >
< div class = "line" > < a name = "l00579" > < / a > < span class = "lineno" > 579< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00580" > < / a > < span class = "lineno" > 580< / span >   < span class = "comment" > /// Use the texture cache for reads; cast as a particular type< / span > < / div >
< div class = "line" > < a name = "l00581" > < / a > < span class = "lineno" > 581< / span >   < span class = "comment" > < / span > < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T> < / div >
< div class = "line" > < a name = "l00582" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a3f29fab81a72a8bdd93901851af98ec7" > 582< / a > < / span >   __device__ < span class = "keyword" > inline< / span > T < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a3f29fab81a72a8bdd93901851af98ec7" > ldgAs< / a > ()< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00583" > < / a > < span class = "lineno" > 583< / span >   < span class = "preprocessor" > #if __CUDA_ARCH__ > = 350< / span > < / div >
< div class = "line" > < a name = "l00584" > < / a > < span class = "lineno" > 584< / span >   < span class = "preprocessor" > < / span > < span class = "keywordflow" > return< / span > __ldg(dataAs< T> ());< / div >
< div class = "line" > < a name = "l00585" > < / a > < span class = "lineno" > 585< / span >   < span class = "preprocessor" > #else< / span > < / div >
< div class = "line" > < a name = "l00586" > < / a > < span class = "lineno" > 586< / span >   < span class = "preprocessor" > < / span > < span class = "keywordflow" > return< / span > as< T> ();< / div >
< div class = "line" > < a name = "l00587" > < / a > < span class = "lineno" > 587< / span >   < span class = "preprocessor" > #endif< / span > < / div >
< div class = "line" > < a name = "l00588" > < / a > < span class = "lineno" > 588< / span >   < span class = "preprocessor" > < / span > }< / div >
< div class = "line" > < a name = "l00589" > < / a > < span class = "lineno" > 589< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00590" > < / a > < span class = "lineno" > 590< / span >   < span class = "comment" > /// Returns a tensor that is a view of the SubDim-dimensional slice< / span > < / div >
< div class = "line" > < a name = "l00591" > < / a > < span class = "lineno" > 591< / span >   < span class = "comment" > /// of this tensor, starting where our data begins< / span > < / div >
< div class = "line" > < a name = "l00592" > < / a > < span class = "lineno" > 592< / span >   < span class = "comment" > < / span > < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< / a > < < span class = "keyword" > typename< / span > TensorType::DataType,< / div >
< div class = "line" > < a name = "l00593" > < / a > < span class = "lineno" > 593< / span >   SubDim,< / div >
< div class = "line" > < a name = "l00594" > < / a > < span class = "lineno" > 594< / span >   TensorType::IsContig,< / div >
< div class = "line" > < a name = "l00595" > < / a > < span class = "lineno" > 595< / span >   < span class = "keyword" > typename< / span > TensorType::IndexType,< / div >
< div class = "line" > < a name = "l00596" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a0ea642a74fc7070130a4a20635dc7b3c" > 596< / a > < / span >   PtrTraits> < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a0ea642a74fc7070130a4a20635dc7b3c" > view< / a > () {< / div >
< div class = "line" > < a name = "l00597" > < / a > < span class = "lineno" > 597< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a2d4e62fb08c180dfe2bde8d47361d61f" > tensor_< / a > .template view< SubDim> (< a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > );< / div >
< div class = "line" > < a name = "l00598" > < / a > < span class = "lineno" > 598< / span >   }< / div >
< div class = "line" > < a name = "l00599" > < / a > < span class = "lineno" > 599< / span >   < / div >
< div class = "line" > < a name = "l00600" > < / a > < span class = "lineno" > 600< / span >   < span class = "keyword" > protected< / span > :< span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00601" > < / a > < span class = "lineno" > 601< / span >   < span class = "comment" > /// One dimension greater can create us< / span > < / div >
< div class = "line" > < a name = "l00602" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a33929e7387099e4e49be139ba467ebfc" > 602< / a > < / span >   < span class = "comment" > < / span > < span class = "keyword" > friend< / span > < span class = "keyword" > class < / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > < TensorType, SubDim + 1, PtrTraits> ;< / div >
< div class = "line" > < a name = "l00603" > < / a > < span class = "lineno" > 603< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00604" > < / a > < span class = "lineno" > 604< / span >   < span class = "comment" > /// Our parent tensor can create us< / span > < / div >
< div class = "line" > < a name = "l00605" > < / a > < span class = "lineno" > 605< / span >   < span class = "comment" > < / span > < span class = "keyword" > friend< / span > < span class = "keyword" > class< / span > < / div >
< div class = "line" > < a name = "l00606" > < / a > < span class = "lineno" > 606< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > Tensor< / a > < < span class = "keyword" > typename< / span > TensorType::DataType,< / div >
< div class = "line" > < a name = "l00607" > < / a > < span class = "lineno" > 607< / span >   TensorType::NumDim,< / div >
< div class = "line" > < a name = "l00608" > < / a > < span class = "lineno" > 608< / span >   TensorType::IsContig,< / div >
< div class = "line" > < a name = "l00609" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#ae095aab5ad40d0b021d5cdf8bea83f67" > 609< / a > < / span >   < span class = "keyword" > typename< / span > TensorType::IndexType,< / div >
< div class = "line" > < a name = "l00610" > < / a > < span class = "lineno" > 610< / span >   PtrTraits> ;< / div >
< div class = "line" > < a name = "l00611" > < / a > < span class = "lineno" > 611< / span >   < / div >
< div class = "line" > < a name = "l00612" > < / a > < span class = "lineno" > 612< / span >   __host__ __device__ < span class = "keyword" > inline< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > SubTensor< / a > (< / div >
< div class = "line" > < a name = "l00613" > < / a > < span class = "lineno" > 613< / span >   TensorType& t,< / div >
< div class = "line" > < a name = "l00614" > < / a > < span class = "lineno" > 614< / span >   < span class = "keyword" > typename< / span > TensorType::DataPtrType < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a30dff4e7bea94cd894e17f6bdd7a7eb1" > data< / a > )< / div >
< div class = "line" > < a name = "l00615" > < / a > < span class = "lineno" > 615< / span >   : < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a2d4e62fb08c180dfe2bde8d47361d61f" > tensor_< / a > (t),< / div >
< div class = "line" > < a name = "l00616" > < / a > < span class = "lineno" > 616< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > (data) {< / div >
< div class = "line" > < a name = "l00617" > < / a > < span class = "lineno" > 617< / span >   }< / div >
< div class = "line" > < a name = "l00618" > < / a > < span class = "lineno" > 618< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00619" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a2d4e62fb08c180dfe2bde8d47361d61f" > 619< / a > < / span >   < span class = "comment" > /// The tensor we' re referencing< / span > < / div >
< div class = "line" > < a name = "l00620" > < / a > < span class = "lineno" > 620< / span >   < span class = "comment" > < / span > TensorType& < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a2d4e62fb08c180dfe2bde8d47361d61f" > tensor_< / a > ;< / div >
< div class = "line" > < a name = "l00621" > < / a > < span class = "lineno" > 621< / span >   < span class = "comment" > < / span > < / div >
< div class = "line" > < a name = "l00622" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > 622< / a > < / span >   < span class = "comment" > /// The start of our sub-region< / span > < / div >
< div class = "line" > < a name = "l00623" > < / a > < span class = "lineno" > 623< / span >   < span class = "comment" > < / span > < span class = "keyword" > typename< / span > TensorType::DataPtrType < span class = "keyword" > const< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > data_< / a > ;< / div >
< div class = "line" > < a name = "l00624" > < / a > < span class = "lineno" > 624< / span >   };< / div >
< div class = "line" > < a name = "l00625" > < / a > < span class = "lineno" > 625< / span >   < / div >
< div class = "line" > < a name = "l00626" > < / a > < span class = "lineno" > 626< / span >   } < span class = "comment" > // namespace detail< / span > < / div >
< div class = "line" > < a name = "l00627" > < / a > < span class = "lineno" > 627< / span >   < / div >
< div class = "line" > < a name = "l00628" > < / a > < span class = "lineno" > 628< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T, < span class = "keywordtype" > int< / span > Dim, < span class = "keywordtype" > bool< / span > Contig,< / div >
< div class = "line" > < a name = "l00629" > < / a > < span class = "lineno" > 629< / span >   < span class = "keyword" > typename< / span > IndexT, < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < span class = "keyword" > class < / span > PtrTraits> < / div >
< div class = "line" > < a name = "l00630" > < / a > < span class = "lineno" > 630< / span >   __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00631" > < / a > < span class = "lineno" > 631< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > detail::SubTensor< Tensor< T, Dim, Contig, IndexT, PtrTraits> < / a > ,< / div >
< div class = "line" > < a name = "l00632" > < / a > < span class = "lineno" > 632< / span >   Dim - 1, PtrTraits> < / div >
< div class = "line" > < a name = "l00633" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#a204ea1c3232b51512a9c027bb44ee3c4" > 633< / a > < / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a204ea1c3232b51512a9c027bb44ee3c4" > Tensor< T, Dim, Contig, IndexT, PtrTraits> ::operator[]< / a > (IndexT index) {< / div >
< div class = "line" > < a name = "l00634" > < / a > < span class = "lineno" > 634< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > detail::SubTensor< / a > < < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > TensorType< / a > , Dim - 1, PtrTraits> (< / div >
< div class = "line" > < a name = "l00635" > < / a > < span class = "lineno" > 635< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > detail::SubTensor< TensorType, Dim, PtrTraits> < / a > (< / div >
< div class = "line" > < a name = "l00636" > < / a > < span class = "lineno" > 636< / span >   *< span class = "keyword" > this< / span > , data_)[index]);< / div >
< div class = "line" > < a name = "l00637" > < / a > < span class = "lineno" > 637< / span >   }< / div >
< div class = "line" > < a name = "l00638" > < / a > < span class = "lineno" > 638< / span >   < / div >
< div class = "line" > < a name = "l00639" > < / a > < span class = "lineno" > 639< / span >   < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > T, < span class = "keywordtype" > int< / span > Dim, < span class = "keywordtype" > bool< / span > Contig,< / div >
< div class = "line" > < a name = "l00640" > < / a > < span class = "lineno" > 640< / span >   < span class = "keyword" > typename< / span > IndexT, < span class = "keyword" > template< / span > < < span class = "keyword" > typename< / span > U> < span class = "keyword" > class < / span > PtrTraits> < / div >
< div class = "line" > < a name = "l00641" > < / a > < span class = "lineno" > 641< / span >   __host__ __device__ < span class = "keyword" > inline< / span > < / div >
< div class = "line" > < a name = "l00642" > < / a > < span class = "lineno" > 642< / span >   < span class = "keyword" > const< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > detail::SubTensor< Tensor< T, Dim, Contig, IndexT, PtrTraits> < / a > ,< / div >
< div class = "line" > < a name = "l00643" > < / a > < span class = "lineno" > 643< / span >   Dim - 1, PtrTraits> < / div >
< div class = "line" > < a name = "l00644" > < / a > < span class = "lineno" > < a class = "line" href = "classfaiss_1_1gpu_1_1Tensor.html#a6538556caad2fd01bd81ba441185d9f3" > 644< / a > < / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html#a204ea1c3232b51512a9c027bb44ee3c4" > Tensor< T, Dim, Contig, IndexT, PtrTraits> ::operator[]< / a > (IndexT index)< span class = "keyword" > const < / span > {< / div >
< div class = "line" > < a name = "l00645" > < / a > < span class = "lineno" > 645< / span >   < span class = "keywordflow" > return< / span > < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > detail::SubTensor< / a > < < a class = "code" href = "classfaiss_1_1gpu_1_1Tensor.html" > TensorType< / a > , Dim - 1, PtrTraits> (< / div >
< div class = "line" > < a name = "l00646" > < / a > < span class = "lineno" > 646< / span >   < a class = "code" href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > detail::SubTensor< TensorType, Dim, PtrTraits> < / a > (< / div >
< div class = "line" > < a name = "l00647" > < / a > < span class = "lineno" > 647< / span >   < span class = "keyword" > const_cast< < / span > TensorType& < span class = "keyword" > > < / span > (*this), data_)[index]);< / div >
< div class = "line" > < a name = "l00648" > < / a > < span class = "lineno" > 648< / span >   }< / div >
< div class = "line" > < a name = "l00649" > < / a > < span class = "lineno" > 649< / span >   < / div >
< div class = "line" > < a name = "l00650" > < / a > < span class = "lineno" > 650< / span >   } } < span class = "comment" > // namespace< / span > < / div >
< div class = "line" > < a name = "l00651" > < / a > < span class = "lineno" > 651< / span >   < / div >
< div class = "line" > < a name = "l00652" > < / a > < span class = "lineno" > 652< / span >   < span class = "preprocessor" > #include " Tensor-inl.cuh" < / span > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a151fd78dd5695b1346739e32a9e76309" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a151fd78dd5695b1346739e32a9e76309" > faiss::gpu::Tensor::Tensor< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor()< / div > < div class = "ttdoc" > Default constructor. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00021" > Tensor-inl.cuh:21< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_ad02cd32977ebf89636c8d085b4470c2e" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#ad02cd32977ebf89636c8d085b4470c2e" > faiss::gpu::Tensor::dataAs< / a > < / div > < div class = "ttdeci" > __host__ __device__ const PtrTraits< const U > ::PtrType dataAs() const < / div > < div class = "ttdoc" > Cast to a different datatype. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00206" > Tensor.cuh:206< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a4bcfb3f04077a42e0d8c924e11b2d637" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a4bcfb3f04077a42e0d8c924e11b2d637" > faiss::gpu::Tensor::downcastInner< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor< T, NewDim, Contig, IndexT, PtrTraits > downcastInner()< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00536" > Tensor-inl.cuh:536< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_a485abbadb5b5de23e88822366857a78f" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a485abbadb5b5de23e88822366857a78f" > faiss::gpu::detail::SubTensor::dataAs< / a > < / div > < div class = "ttdeci" > __host__ __device__ PtrTraits< const T > ::PtrType dataAs() const < / div > < div class = "ttdoc" > Cast to a different datatype (const) < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00567" > Tensor.cuh:567< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a31075cd6df497e4c1a7c7e47fa2603ca" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a31075cd6df497e4c1a7c7e47fa2603ca" > faiss::gpu::Tensor::isContiguousDim< / a > < / div > < div class = "ttdeci" > __host__ __device__ bool isContiguousDim(int i) const < / div > < div class = "ttdoc" > Returns true if the given dimension index has no padding. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00397" > Tensor-inl.cuh:397< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_a3f29fab81a72a8bdd93901851af98ec7" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a3f29fab81a72a8bdd93901851af98ec7" > faiss::gpu::detail::SubTensor::ldgAs< / a > < / div > < div class = "ttdeci" > __device__ T ldgAs() const < / div > < div class = "ttdoc" > Use the texture cache for reads; cast as a particular type. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00582" > Tensor.cuh:582< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_aa8b48391a291a7e214c7204d81cd9d73" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#aa8b48391a291a7e214c7204d81cd9d73" > faiss::gpu::Tensor::transpose< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor< T, Dim, Contig, IndexT, PtrTraits > transpose(int dim1, int dim2) const < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00406" > Tensor-inl.cuh:406< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a37d3ac0dffcaad29f09d6e85fb07b335" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a37d3ac0dffcaad29f09d6e85fb07b335" > faiss::gpu::Tensor::data< / a > < / div > < div class = "ttdeci" > __host__ __device__ DataPtrType data()< / div > < div class = "ttdoc" > Returns a raw pointer to the start of our data. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00174" > Tensor.cuh:174< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_a9825bed624c3abb6337a1ab7654d7db7" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a9825bed624c3abb6337a1ab7654d7db7" > faiss::gpu::detail::SubTensor::dataAs< / a > < / div > < div class = "ttdeci" > __host__ __device__ PtrTraits< T > ::PtrType dataAs()< / div > < div class = "ttdoc" > Cast to a different datatype. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00560" > Tensor.cuh:560< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a782cd978394660371a22229f02e49e60" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a782cd978394660371a22229f02e49e60" > faiss::gpu::Tensor::narrowOutermost< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor< T, Dim, Contig, IndexT, PtrTraits > narrowOutermost(IndexT start, IndexT size)< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00606" > Tensor-inl.cuh:606< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4_html_a0ec3a48f265de627490e7cdf540e9fc5" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a0ec3a48f265de627490e7cdf540e9fc5" > faiss::gpu::detail::SubTensor< TensorType, 0, PtrTraits > ::dataAs< / a > < / div > < div class = "ttdeci" > __host__ __device__ PtrTraits< const T > ::PtrType dataAs() const < / div > < div class = "ttdoc" > Cast to a different datatype (const) < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00438" > Tensor.cuh:438< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a08024dbcaed3c1235cac62b476f14509" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a08024dbcaed3c1235cac62b476f14509" > faiss::gpu::Tensor::data_< / a > < / div > < div class = "ttdeci" > DataPtrType data_< / div > < div class = "ttdoc" > Raw pointer to where the tensor data begins. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00343" > Tensor.cuh:343< / a > < / div > < / div >
2017-02-23 06:26:44 +08:00
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a55896db55a03e43a1789886aa22a0ba4" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a55896db55a03e43a1789886aa22a0ba4" > faiss::gpu::Tensor::operator=< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor< T, Dim, Contig, IndexT, PtrTraits > & operator=(Tensor< T, Dim, Contig, IndexT, PtrTraits > & t)=default< / div > < div class = "ttdoc" > Assignment. < / div > < / div >
2017-03-21 01:33:21 +08:00
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_a6e7097578ba17c10895ec0dafa385901" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a6e7097578ba17c10895ec0dafa385901" > faiss::gpu::detail::SubTensor::data< / a > < / div > < div class = "ttdeci" > __host__ __device__ const TensorType::DataPtrType data() const < / div > < div class = "ttdoc" > Returns a raw accessor to our slice (const). < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00541" > Tensor.cuh:541< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4_html_ac44400045b113c527d6ed59a910f885c" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#ac44400045b113c527d6ed59a910f885c" > faiss::gpu::detail::SubTensor< TensorType, 0, PtrTraits > ::ldg< / a > < / div > < div class = "ttdeci" > __device__ TensorType::DataType ldg() const < / div > < div class = "ttdoc" > Use the texture cache for reads. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00443" > Tensor.cuh:443< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_ad6df1805b76f74018bc94fffab8987ac" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#ad6df1805b76f74018bc94fffab8987ac" > faiss::gpu::Tensor::upcastOuter< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor< T, NewDim, Contig, IndexT, PtrTraits > upcastOuter()< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00435" > Tensor-inl.cuh:435< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a080cf59b73bee015c02947009e70ce07" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a080cf59b73bee015c02947009e70ce07" > faiss::gpu::Tensor::canCastIndexType< / a > < / div > < div class = "ttdeci" > __host__ bool canCastIndexType() const < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00307" > Tensor-inl.cuh:307< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4_html_a67bfa92466e03834b7f007cb9cdf8d50" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a67bfa92466e03834b7f007cb9cdf8d50" > faiss::gpu::detail::SubTensor< TensorType, 0, PtrTraits > ::as< / a > < / div > < div class = "ttdeci" > __host__ __device__ T & as()< / div > < div class = "ttdoc" > Cast to a different datatype. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00418" > Tensor.cuh:418< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_a0d32586e8f6f22f5f90bca566d901d0b" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a0d32586e8f6f22f5f90bca566d901d0b" > faiss::gpu::detail::SubTensor::as< / a > < / div > < div class = "ttdeci" > __host__ __device__ T & as()< / div > < div class = "ttdoc" > Cast to a different datatype. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00547" > Tensor.cuh:547< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4_html_a6e0f585a739cd1474ec24f56609d6501" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a6e0f585a739cd1474ec24f56609d6501" > faiss::gpu::detail::SubTensor< TensorType, 0, PtrTraits > ::tensor_< / a > < / div > < div class = "ttdeci" > TensorType & tensor_< / div > < div class = "ttdoc" > The tensor we& #39;re referencing. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00480" > Tensor.cuh:480< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a58d7a953431def329bac11374679c2dd" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a58d7a953431def329bac11374679c2dd" > faiss::gpu::Tensor::narrow< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor< T, Dim, false, IndexT, PtrTraits > narrow(int dim, IndexT start, IndexT size)< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00630" > Tensor-inl.cuh:630< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_a760782118b71504348d073ca1c92843a" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a760782118b71504348d073ca1c92843a" > faiss::gpu::detail::SubTensor::ldg< / a > < / div > < div class = "ttdeci" > __device__ TensorType::DataType ldg() const < / div > < div class = "ttdoc" > Use the texture cache for reads. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00572" > Tensor.cuh:572< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_ab462574aa0756b701e52279c13449764" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#ab462574aa0756b701e52279c13449764" > faiss::gpu::Tensor::cast< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor< U, Dim, Contig, IndexT, PtrTraits > cast()< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00204" > Tensor-inl.cuh:204< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a6ce60d784817469d5b47b560da177b92" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a6ce60d784817469d5b47b560da177b92" > faiss::gpu::Tensor::copyTo< / a > < / div > < div class = "ttdeci" > __host__ void copyTo(Tensor< T, Dim, Contig, IndexT, PtrTraits > & t, cudaStream_t stream)< / div > < div class = "ttdoc" > Copies ourselves into a tensor; sizes must match. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00140" > Tensor-inl.cuh:140< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_aa4aa193f6140219872839ad8665b5d36" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aa4aa193f6140219872839ad8665b5d36" > faiss::gpu::detail::SubTensor::data_< / a > < / div > < div class = "ttdeci" > TensorType::DataPtrType const data_< / div > < div class = "ttdoc" > The start of our sub-region. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00622" > Tensor.cuh:622< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a013fac5bd3c78e90b35a6f46dbf3ffda" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a013fac5bd3c78e90b35a6f46dbf3ffda" > faiss::gpu::Tensor::isSame< / a > < / div > < div class = "ttdeci" > __host__ __device__ bool isSame(const Tensor< T, OtherDim, Contig, IndexT, PtrTraits > & rhs) const < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00179" > Tensor-inl.cuh:179< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_afa66c4592bc873a3f663d765cb9ea2c0" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#afa66c4592bc873a3f663d765cb9ea2c0" > faiss::gpu::Tensor::castIndexType< / a > < / div > < div class = "ttdeci" > __host__ Tensor< T, Dim, Contig, NewIndexT, PtrTraits > castIndexType() const < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00287" > Tensor-inl.cuh:287< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a41f7f08246bf8584929693bb1896205f" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a41f7f08246bf8584929693bb1896205f" > faiss::gpu::Tensor::numElements< / a > < / div > < div class = "ttdeci" > __host__ __device__ IndexT numElements() const < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00338" > Tensor-inl.cuh:338< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4_html_aa56767066e40a4758e37b26e43449f1d" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#aa56767066e40a4758e37b26e43449f1d" > faiss::gpu::detail::SubTensor< TensorType, 0, PtrTraits > ::ldgAs< / a > < / div > < div class = "ttdeci" > __device__ T ldgAs() const < / div > < div class = "ttdoc" > Use the texture cache for reads; cast as a particular type. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00453" > Tensor.cuh:453< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_a4c3006fcd82c301b11505620e3e96378" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a4c3006fcd82c301b11505620e3e96378" > faiss::gpu::detail::SubTensor::operator[]< / a > < / div > < div class = "ttdeci" > __host__ __device__ const SubTensor< TensorType, SubDim-1, PtrTraits > operator[](typename TensorType::IndexType index) const < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00512" > Tensor.cuh:512< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_a0ea642a74fc7070130a4a20635dc7b3c" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a0ea642a74fc7070130a4a20635dc7b3c" > faiss::gpu::detail::SubTensor::view< / a > < / div > < div class = "ttdeci" > Tensor< typename TensorType::DataType, SubDim, TensorType::IsContig, typename TensorType::IndexType, PtrTraits > view()< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00596" > Tensor.cuh:596< / a > < / div > < / div >
2017-02-23 06:26:44 +08:00
< div class = "ttc" id = "structfaiss_1_1gpu_1_1traits_1_1RestrictPtrTraits_html" > < div class = "ttname" > < a href = "structfaiss_1_1gpu_1_1traits_1_1RestrictPtrTraits.html" > faiss::gpu::traits::RestrictPtrTraits< / a > < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00044" > Tensor.cuh:44< / a > < / div > < / div >
2017-03-21 01:33:21 +08:00
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4_html_a6a548d4edb57d072be52cd827f055d6d" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a6a548d4edb57d072be52cd827f055d6d" > faiss::gpu::detail::SubTensor< TensorType, 0, PtrTraits > ::data_< / a > < / div > < div class = "ttdeci" > TensorType::DataPtrType const data_< / div > < div class = "ttdoc" > Where our value is located. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00483" > Tensor.cuh:483< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a9a0a293f287e9cba68f31c9e57b79c64" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a9a0a293f287e9cba68f31c9e57b79c64" > faiss::gpu::Tensor::strides< / a > < / div > < div class = "ttdeci" > __host__ __device__ const IndexT * strides() const < / div > < div class = "ttdoc" > Returns the stride array. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00248" > Tensor.cuh:248< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a57855c6e0c529a309168833017049ab5" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a57855c6e0c529a309168833017049ab5" > faiss::gpu::Tensor::getStride< / a > < / div > < div class = "ttdeci" > __host__ __device__ IndexT getStride(int i) const < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00228" > Tensor.cuh:228< / a > < / div > < / div >
2017-02-23 06:26:44 +08:00
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html" > faiss::gpu::Tensor< / a > < / div > < div class = "ttdoc" > Our tensor type. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00031" > Tensor.cuh:31< / a > < / div > < / div >
2017-03-21 01:33:21 +08:00
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_ac3e0306267da475d8140bf6d12cf4af6" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#ac3e0306267da475d8140bf6d12cf4af6" > faiss::gpu::Tensor::sizes< / a > < / div > < div class = "ttdeci" > __host__ __device__ const IndexT * sizes() const < / div > < div class = "ttdoc" > Returns the size array. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00243" > Tensor.cuh:243< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a7c987316708113dafd11971092d59f69" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a7c987316708113dafd11971092d59f69" > faiss::gpu::Tensor::dataAs< / a > < / div > < div class = "ttdeci" > __host__ __device__ PtrTraits< U > ::PtrType dataAs()< / div > < div class = "ttdoc" > Cast to a different datatype. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00199" > Tensor.cuh:199< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_aae42c9c020682fd094958976d8e610ad" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#aae42c9c020682fd094958976d8e610ad" > faiss::gpu::Tensor::getSizeInBytes< / a > < / div > < div class = "ttdeci" > __host__ __device__ size_t getSizeInBytes() const < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00238" > Tensor.cuh:238< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_ad35e0c816162fd08e372b21b79cab6c1" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#ad35e0c816162fd08e372b21b79cab6c1" > faiss::gpu::Tensor::end< / a > < / div > < div class = "ttdeci" > __host__ __device__ DataPtrType end()< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00180" > Tensor.cuh:180< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4_html" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html" > faiss::gpu::detail::SubTensor< TensorType, 0, PtrTraits > < / a > < / div > < div class = "ttdoc" > Specialization for a view of a single value (0-dimensional) < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00377" > Tensor.cuh:377< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a70308137b566362d4cc37d3e71b3ae98" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a70308137b566362d4cc37d3e71b3ae98" > faiss::gpu::Tensor::upcastInner< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor< T, NewDim, Contig, IndexT, PtrTraits > upcastInner()< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00464" > Tensor-inl.cuh:464< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a67ccf916f78dacf0ae0d4816a663eb3f" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a67ccf916f78dacf0ae0d4816a663eb3f" > faiss::gpu::Tensor::view< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor< T, SubDim, Contig, IndexT, PtrTraits > view()< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00599" > Tensor-inl.cuh:599< / a > < / div > < / div >
2017-02-23 06:26:44 +08:00
< div class = "ttc" id = "structfaiss_1_1gpu_1_1traits_1_1DefaultPtrTraits_html" > < div class = "ttname" > < a href = "structfaiss_1_1gpu_1_1traits_1_1DefaultPtrTraits.html" > faiss::gpu::traits::DefaultPtrTraits< / a > < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00049" > Tensor.cuh:49< / a > < / div > < / div >
2017-03-21 01:33:21 +08:00
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_a2d4e62fb08c180dfe2bde8d47361d61f" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a2d4e62fb08c180dfe2bde8d47361d61f" > faiss::gpu::detail::SubTensor::tensor_< / a > < / div > < div class = "ttdeci" > TensorType & tensor_< / div > < div class = "ttdoc" > The tensor we& #39;re referencing. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00619" > Tensor.cuh:619< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a00d2d17504f63b9e834afac730a64324" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a00d2d17504f63b9e834afac730a64324" > faiss::gpu::Tensor::end< / a > < / div > < div class = "ttdeci" > __host__ __device__ DataPtrType end() const < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00192" > Tensor.cuh:192< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4_html_a750047ff919799af43b4861b580c82e3" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a750047ff919799af43b4861b580c82e3" > faiss::gpu::detail::SubTensor< TensorType, 0, PtrTraits > ::data< / a > < / div > < div class = "ttdeci" > __host__ __device__ const TensorType::DataPtrType data() const < / div > < div class = "ttdoc" > Returns a raw accessor to our slice (const). < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00412" > Tensor.cuh:412< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_ac722ca465d06da122898a07ce38276e2" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#ac722ca465d06da122898a07ce38276e2" > faiss::gpu::detail::SubTensor::operator[]< / a > < / div > < div class = "ttdeci" > __host__ __device__ SubTensor< TensorType, SubDim-1, PtrTraits > operator[](typename TensorType::IndexType index)< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00496" > Tensor.cuh:496< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a2d276c97faf432cdc9f3552da63c0d3c" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a2d276c97faf432cdc9f3552da63c0d3c" > faiss::gpu::Tensor::getSize< / a > < / div > < div class = "ttdeci" > __host__ __device__ IndexT getSize(int i) const < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00222" > Tensor.cuh:222< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_ae981a94263044f38be89d690dd958426" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#ae981a94263044f38be89d690dd958426" > faiss::gpu::Tensor::copyFrom< / a > < / div > < div class = "ttdeci" > __host__ void copyFrom(Tensor< T, Dim, Contig, IndexT, PtrTraits > & t, cudaStream_t stream)< / div > < div class = "ttdoc" > Copies a tensor into ourselves; sizes must match. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00102" > Tensor-inl.cuh:102< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_aefdafcf236e5c49ad3bce1646797f8f2" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#aefdafcf236e5c49ad3bce1646797f8f2" > faiss::gpu::detail::SubTensor::as< / a > < / div > < div class = "ttdeci" > __host__ __device__ const T & as() const < / div > < div class = "ttdoc" > Cast to a different datatype (const). < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00553" > Tensor.cuh:553< / a > < / div > < / div >
2017-02-23 06:26:44 +08:00
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html" > faiss::gpu::detail::SubTensor< / a > < / div > < div class = "ttdoc" > A SubDim-rank slice of a parent Tensor. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00038" > Tensor.cuh:38< / a > < / div > < / div >
2017-03-21 01:33:21 +08:00
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4_html_a23e80555a443797d60ae16d605dacd23" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#a23e80555a443797d60ae16d605dacd23" > faiss::gpu::detail::SubTensor< TensorType, 0, PtrTraits > ::dataAs< / a > < / div > < div class = "ttdeci" > __host__ __device__ PtrTraits< T > ::PtrType dataAs()< / div > < div class = "ttdoc" > Cast to a different datatype. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00431" > Tensor.cuh:431< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_html_a30dff4e7bea94cd894e17f6bdd7a7eb1" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor.html#a30dff4e7bea94cd894e17f6bdd7a7eb1" > faiss::gpu::detail::SubTensor::data< / a > < / div > < div class = "ttdeci" > __host__ __device__ TensorType::DataPtrType data()< / div > < div class = "ttdoc" > Returns a raw accessor to our slice. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00535" > Tensor.cuh:535< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a46c20e6be3ce16feb99f8d3a5ee268c9" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a46c20e6be3ce16feb99f8d3a5ee268c9" > faiss::gpu::Tensor::stride_< / a > < / div > < div class = "ttdeci" > IndexT stride_[Dim]< / div > < div class = "ttdoc" > Array of strides (in sizeof(T) terms) per each dimension. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00346" > Tensor.cuh:346< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_acd445331bcc0373cf925f7a02660107c" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#acd445331bcc0373cf925f7a02660107c" > faiss::gpu::Tensor::isContiguous< / a > < / div > < div class = "ttdeci" > __host__ __device__ bool isContiguous() const < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00351" > Tensor-inl.cuh:351< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a5591c571ac52482fc7ceea6ef99af794" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a5591c571ac52482fc7ceea6ef99af794" > faiss::gpu::Tensor::downcastOuter< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor< T, NewDim, Contig, IndexT, PtrTraits > downcastOuter()< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00491" > Tensor-inl.cuh:491< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_afe1bd60ecb861eabafaad7362a68ec1f" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#afe1bd60ecb861eabafaad7362a68ec1f" > faiss::gpu::Tensor::size_< / a > < / div > < div class = "ttdeci" > IndexT size_[Dim]< / div > < div class = "ttdoc" > Size per each dimension. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00349" > Tensor.cuh:349< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4_html_aae8c90b402493f5656f94701157a7417" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#aae8c90b402493f5656f94701157a7417" > faiss::gpu::detail::SubTensor< TensorType, 0, PtrTraits > ::data< / a > < / div > < div class = "ttdeci" > __host__ __device__ TensorType::DataPtrType data()< / div > < div class = "ttdoc" > Returns a raw accessor to our slice. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00406" > Tensor.cuh:406< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a204ea1c3232b51512a9c027bb44ee3c4" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a204ea1c3232b51512a9c027bb44ee3c4" > faiss::gpu::Tensor::operator[]< / a > < / div > < div class = "ttdeci" > __host__ __device__ detail::SubTensor< TensorType, Dim-1, PtrTraits > operator[](IndexT)< / div > < div class = "ttdoc" > Returns a read/write view of a portion of our tensor. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00633" > Tensor.cuh:633< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a9b3a5fb3a903295b7d2f8cb2caae0352" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a9b3a5fb3a903295b7d2f8cb2caae0352" > faiss::gpu::Tensor::canCastResize< / a > < / div > < div class = "ttdeci" > __host__ __device__ bool canCastResize() const < / div > < div class = "ttdoc" > Returns true if we can castResize() this tensor to the new type. < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00260" > Tensor-inl.cuh:260< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4_html_ad1d375e64756991dadeb5a1e63ed2cfd" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1detail_1_1SubTensor_3_01TensorType_00_010_00_01PtrTraits_01_4.html#ad1d375e64756991dadeb5a1e63ed2cfd" > faiss::gpu::detail::SubTensor< TensorType, 0, PtrTraits > ::as< / a > < / div > < div class = "ttdeci" > __host__ __device__ const T & as() const < / div > < div class = "ttdoc" > Cast to a different datatype (const). < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00424" > Tensor.cuh:424< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_a9a098dad7e211328d1ae8b00d02839ce" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#a9a098dad7e211328d1ae8b00d02839ce" > faiss::gpu::Tensor::data< / a > < / div > < div class = "ttdeci" > __host__ __device__ const DataPtrType data() const < / div > < div class = "ttdoc" > Returns a raw pointer to the start of our data (const). < / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor_8cuh_source.html#l00186" > Tensor.cuh:186< / a > < / div > < / div >
< div class = "ttc" id = "classfaiss_1_1gpu_1_1Tensor_html_aab60b0711646ffe587b62a5434d53c6a" > < div class = "ttname" > < a href = "classfaiss_1_1gpu_1_1Tensor.html#aab60b0711646ffe587b62a5434d53c6a" > faiss::gpu::Tensor::castResize< / a > < / div > < div class = "ttdeci" > __host__ __device__ Tensor< U, Dim, Contig, IndexT, PtrTraits > castResize()< / div > < div class = "ttdef" > < b > Definition:< / b > < a href = "Tensor-inl_8cuh_source.html#l00226" > Tensor-inl.cuh:226< / a > < / div > < / div >
2017-02-23 06:26:44 +08:00
< / div > <!-- fragment --> < / div > <!-- contents -->
<!-- start footer part -->
< hr class = "footer" / > < address class = "footer" > < small >
Generated by   < a href = "http://www.doxygen.org/index.html" >
< img class = "footer" src = "doxygen.png" alt = "doxygen" / >
< / a > 1.8.5
< / small > < / address >
< / body >
< / html >
