Lines Matching defs:shift
46 int offset = 0; // the shift along the flattened tensor for current element
53 const int shift = dim_size[i] - threshold[i];
56 // calculate dimension index after the shift
57 const int shifted_indx = (indx + shift) % dim_size[i];
70 // dim_range[j] = threshold[j] + shift[j]
71 // offset = shift[j] + ... other offsets
78 } else if (threshold[j] != 0) { // if threshold is 0 shift is 0
98 // isd - inner shift dimension
140 const int shift = dim_size[i] - threshold[i];
143 // calculate dimension index after the shift
144 int out_indx = (indx + shift) % dim_size[i];
174 // shift i and the pointers over to the next group position
181 // the isd (inner shift dim) should skip to next threshold or endpoint
197 } else if (threshold[j] != 0) { // if threshold is 0 shift is 0
230 const Tensor& shift = context->input(1);
233 auto shift_flat = shift.flat<Tshift>();
238 OP_REQUIRES(context, shift.shape().dims() <= 1,
240 "shift must be a scalar or a 1-D vector. Found: ",
241 shift.shape().DebugString()));
247 context, shift.shape() == axis.shape(),
248 errors::InvalidArgument("shift and axis must have the same size"));
274 // inner shift dimension (inner most shifted dimension)