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.cc26 int64 n_minor_dims = start.size(); in SliceInMinorDims() local
31 TF_RET_CHECK(n_minor_dims <= n_dims); in SliceInMinorDims()
35 /*len=*/n_dims - n_minor_dims); in SliceInMinorDims()
75 const int64 n_minor_dims = start.size(); in UpdateSliceInMinorDims() local
76 TF_RET_CHECK(n_minor_dims <= n_dims); in UpdateSliceInMinorDims()
79 padded_start.begin() + (n_dims - n_minor_dims)); in UpdateSliceInMinorDims()
115 int64 n_minor_dims = starts.size(); in DynamicSliceInMinorDims() local
116 TF_RET_CHECK(n_minor_dims == sizes.size()); in DynamicSliceInMinorDims()
117 TF_RET_CHECK(n_minor_dims <= n_dims); in DynamicSliceInMinorDims()