Home
last modified time | relevance | path

Searched defs:batch_strides (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgather_nd_op_gpu.cu.cc32 const Eigen::array<int64, IXDIM> batch_strides, in GatherSliceOpKernel()
77 Eigen::array<int64, IXDIM> batch_strides; in operator ()() local
Dscatter_nd_op_gpu.cu.cc85 const Eigen::array<int64, IXDIM> batch_strides, const int64 num_indices, in ScatterNdOpKernel()
128 Eigen::array<int64, IXDIM> batch_strides; in operator ()() local