Searched defs:batch_strides (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_nd_op_gpu.cu.cc | 32 T* __restrict__ out, const Eigen::array<int64, IXDIM> batch_strides, in GatherSliceOpKernel()
77 Eigen::array<int64, IXDIM> batch_strides; in operator ()() local
|
| D | scatter_nd_op_gpu.cu.cc | 101 const Eigen::array<int64, IXDIM> batch_strides, const int64 num_indices, in ScatterNdOpKernel()
144 Eigen::array<int64, IXDIM> batch_strides; in operator ()() local
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | LegacyBatchingRegistrations.cpp | 388 auto batch_strides = physical_tensor.strides().slice(0, num_batch_dims); in as_strided_batching_rule() local
|
| /external/pytorch/aten/src/ATen/ |
| D | LegacyBatchingRegistrations.cpp | 640 auto batch_strides = physical_tensor.strides().slice(0, num_batch_dims); in as_strided_batching_rule() local
|