Home
last modified time | relevance | path

Searched full:maximum_indices (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_meta.py1622 output, offset2bag, bag_size, maximum_indices = torch.ops.aten._embedding_bag.default(
1629 grad, indices, offset2bag, bag_size, maximum_indices, weight.size(0),
1634 maximum_indices.to('meta'), weight.size(0),
1650 output, offset2bag, bag_size, maximum_indices = torch.ops.aten._embedding_bag.default(
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNonSymbolicBC.h19 …Tensor & offset2bag, const at::Tensor & bag_size, const at::Tensor & maximum_indices, int64_t num_…
H A Dnative_functions.yaml2339 … indices, Tensor offsets, Tensor offset2bag, Tensor bag_size, Tensor maximum_indices, SymInt num_w…
2347 …sor grad, Tensor indices, Tensor offset2bag, Tensor bag_size, Tensor maximum_indices, SymInt num_w…
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/generated/
H A Dc_shim_cpu.h23 …TensorHandle offset2bag, AtenTensorHandle bag_size, AtenTensorHandle maximum_indices, int64_t num_…
H A Dc_shim_cuda.h26 …TensorHandle offset2bag, AtenTensorHandle bag_size, AtenTensorHandle maximum_indices, int64_t num_…
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py6355 maximum_indices, argument
6382 maximum_indices,
6397 maximum_indices, argument
6410 torch._check(maximum_indices is not None)
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dderivatives.yaml1996 …sor grad, Tensor indices, Tensor offset2bag, Tensor bag_size, Tensor maximum_indices, SymInt num_w…
2000 maximum_indices: non_differentiable