Home | History | Annotate | Download | only in kernels

Lines Matching defs:slice_size

140                           const int32 slice_size, const int32 first_dim_size,
175       const int slice_size = merged->flat_outer_dims<T>().dimension(1);
199                                   j * slice_size));
208       DynamicStitchGPUImpl<T>(c->eigen_gpu_device(), slice_size, first_dim_size,
240       const int slice_size = merged_flat.dimension(1);
241       const size_t slice_bytes = slice_size * sizeof(T);
247             data.shaped<T, 2>({indices_vec.dimension(0), slice_size});
257             memcpy(merged_base + index * slice_size, data_base + i * slice_size,
261           Eigen::DSizes<Eigen::DenseIndex, 2> sizes(1, slice_size);