Home
last modified time | relevance | path

Searched refs:n_minor_dims (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing.cc46 int64 n_minor_dims = start.size(); in SliceInMinorDims() local
51 TF_RET_CHECK(n_minor_dims <= n_dims); in SliceInMinorDims()
55 /*len=*/n_dims - n_minor_dims); in SliceInMinorDims()
96 const int64 n_minor_dims = start.size(); in UpdateSliceInMinorDims() local
97 TF_RET_CHECK(n_minor_dims <= n_dims); in UpdateSliceInMinorDims()
100 padded_start.begin() + (n_dims - n_minor_dims)); in UpdateSliceInMinorDims()
136 int64 n_minor_dims = starts.size(); in DynamicSliceInMinorDims() local
137 TF_RET_CHECK(n_minor_dims == sizes.size()); in DynamicSliceInMinorDims()
138 TF_RET_CHECK(n_minor_dims <= n_dims); in DynamicSliceInMinorDims()