Searched refs:shift_tensor (Results 1 – 1 of 1) sorted by relevance
687 const Tensor& shift_tensor = MklGetInput(context, kShiftIndex); in Compute() local709 OP_REQUIRES(context, shift_tensor.dims() == 1, in Compute()711 shift_tensor.shape().DebugString())); in Compute()819 const U* shift_tf = shift_tensor.flat<U>().data(); in Compute()