Searched defs:batch_strides (Results 1 – 2 of 2) sorted by relevance
32 T* __restrict__ out, const Eigen::array<int64, IXDIM> batch_strides, in GatherSliceOpKernel()77 Eigen::array<int64, IXDIM> batch_strides; in operator ()() local
101 const Eigen::array<int64, IXDIM> batch_strides, const int64 num_indices, in ScatterNdOpKernel()144 Eigen::array<int64, IXDIM> batch_strides; in operator ()() local