Searched defs:gather_dim_size (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | gather_functor_gpu.cu.h | 36 int64 gather_dim_size, int64 indices_size, in GatherOpKernel()
109 const Index* indices, T* out, int64 gather_dim_size, in LaunchGatherKernel()
|
| H A D | gather_functor_batched_gpu.cu.h | 37 int64 gather_dim_size, int64 indices_size, in GatherOpKernel()
125 int64 gather_dim_size, int64 indices_size, in LaunchGatherKernel()
|
| H A D | gather_op.cc | 125 int64_t gather_dim_size = params.dim_size(axis); in Compute() local
|
| H A D | resource_variable_ops.cc | 750 int64_t gather_dim_size = 1; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
| H A D | array_grad.cc | 660 Output indices, int batch_dims, Output gather_dim_size) { in BatchGatherGrad()
748 Output gather_dim_size = in GatherV2Grad() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | array_grad.py | 618 gather_dim_size): argument
|