Searched refs:along_axis (Results 1 – 1 of 1) sorted by relevance
1852 bool along_axis = true; in IsAlongAxis() local1854 along_axis = along_axis && (list.i(i) == axis[i]); in IsAlongAxis()1856 if (along_axis) return true; in IsAlongAxis()1915 bool along_axis = true; in IsAlongAxis() local1917 along_axis = along_axis && (tensor.flat<int>()(i) == axis[i]); in IsAlongAxis()1919 if (along_axis) return true; in IsAlongAxis()