| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_max_test.cpp | 33 Tensor& max_indices) { in op_max_dim_max()
45 Tensor max_indices = tf_in.zeros({2, 3}); in test_max_out_invalid_dimensions() local
105 Tensor max_indices = tfl.zeros(out_shape, dynamism); in test_dynamic_shape() local
131 Tensor max_indices = tf_long.zeros({2, 4}); in test_max_out_dtype() local
194 Tensor max_indices = tf_long.zeros({2, 3, 1}); in test_max_out_dtype() local
302 Tensor max_indices = tf_long.zeros({2, 3, 1}); in TEST_F() local
341 Tensor max_indices = tf_long.zeros({2, 3, 1}); in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | EmbeddingBag.cu | 66 index_t *bag_size, index_t *max_indices, in EmbeddingBag_updateOutputKernel_max()
241 const index_t *max_indices, const scalar_t *gradOutput, in EmbeddingBag_accGradParametersKernel_max()
269 const Tensor &max_indices, in embedding_bag_backward_cuda_max()
384 Tensor max_indices; in _embedding_bag_cuda() local
430 const Tensor &max_indices, in _embedding_bag_dense_backward_cuda()
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/freedreno/a5xx/ |
| H A D | fd5_draw.h | 28 enum a4xx_index_size idx_type, uint32_t max_indices, in fd5_draw()
72 uint32_t max_indices, idx_offset; in fd5_draw_emit() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | EmbeddingBag.cpp | 1017 Tensor max_indices = at::empty(bag_size.sizes(), offsets.options()); in make_max_indices() local
1065 Tensor* max_indices, in embedding_bag_cpu_max_out()
1129 Tensor& bag_size, Tensor* max_indices, in _embedding_bag_cpu_impl_out()
1212 …Tensor max_indices = make_max_indices(weight, indices, offsets, bag_size, mode, include_last_offse… in _embedding_bag_cpu_impl() local
1448 const Tensor& max_indices, in _embedding_bag_dense_backward_cpu_max()
|
| H A D | TensorCompare.cpp | 667 Tensor max_indices = at::empty({0}, self.options().dtype(kLong)); in qmax() local
792 …Tensor&> max_out(const Tensor& self, Dimname dim, bool keepdim, Tensor& max, Tensor& max_indices) { in max_out()
|
| /aosp_15_r20/external/mesa3d/src/gallium/auxiliary/draw/ |
| H A D | draw_vbuf.h | 59 unsigned max_indices; member
|
| H A D | draw_pipe_vbuf.c | 73 unsigned max_indices; member
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/freedreno/a4xx/ |
| H A D | fd4_draw.h | 34 enum a4xx_index_size idx_type, uint32_t max_indices, in fd4_draw()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_max.cpp | 39 Tensor& max_indices) { in max_out()
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/freedreno/a6xx/ |
| H A D | fd6_draw.cc | 82 max_indices(const struct pipe_draw_info *info, unsigned index_offset) in max_indices() function
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | reduce_util.cpp | 429 Tensor& max_indices) { in check_min_max_args()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensor.cpp | 411 Tensor max_indices = in _validate_sparse_coo_tensor_args() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | gather_scatter_handler.cc | 94 std::vector<HloInstruction*> max_indices; in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims() local
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 7124 def _gather(shape, index_dim, max_indices): argument
7152 def _gather(shape, index_dim, max_indices): argument
24426 def index_variable(shape, max_indices, device=torch.device('cpu')): argument
24432 def gather_variable(shape, index_dim, max_indices, duplicate=False, device=torch.device('cpu')): argument
|
| /aosp_15_r20/external/mesa3d/src/microsoft/vulkan/ |
| H A D | dzn_cmd_buffer.c | 3760 uint32_t max_indices = cmdbuf->state.ib.view.SizeInBytes / index_size; in dzn_cmd_buffer_triangle_fan_get_max_index_buf_size() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_autograd.py | 10293 def index_perm_variable(shape, max_indices): argument
|