| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_gather_test.cpp | 31 bool sparse_grad, in op_gather_out()
53 bool sparse_grad = false; in test_gather_out() local
152 bool sparse_grad = false; in test_gather_out_invalid_dim() local
186 bool sparse_grad = false; in test_dynamic_shape() local
214 bool sparse_grad = false; in TEST_F() local
245 bool sparse_grad = false; in TEST_F() local
285 bool sparse_grad = false; in TEST_F() local
297 bool sparse_grad = false; in TEST_F() local
310 bool sparse_grad = false; in TEST_F() local
323 bool sparse_grad = false; in TEST_F() local
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | NamedTensor.cpp | 335 Tensor gather(const Tensor& self, Dimname dim, const Tensor& index, bool sparse_grad) { in gather()
338 Tensor& gather_out(const Tensor& self, Dimname dim, const Tensor& index, bool sparse_grad, Tensor& … in gather_out()
|
| H A D | TensorAdvancedIndexing.cpp | 1649 …kward(const Tensor& grad, const Tensor& self, int64_t dim, const Tensor& index, bool sparse_grad) { in gather_backward()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_gather.cpp | 67 bool sparse_grad, in gather_out()
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | index_util.cpp | 19 bool sparse_grad, in check_gather_args()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_optim.py | 437 def get_grad(param, sparse_grad, w): argument
460 def eval(params, sparse_grad, w): argument
|
| H A D | test_scatter_gather_ops.py | 76 def test_gather_backward_with_empty_index_tensor(self, device, dtype, sparse_grad): argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset11.py | 385 def gather(g: jit_utils.GraphContext, self, dim, index, sparse_grad=False): argument
|
| H A D | symbolic_opset9.py | 5147 def gather(g: jit_utils.GraphContext, self, dim, index, sparse_grad=False): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesScatterOps.cpp | 796 bool sparse_grad) { in gather_batch_rule()
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 2922 def gather(x, dim, index, sparse_grad=False): argument
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 4882 def meta_gather(self, dim, index, sparse_grad=False): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | generated_ops.cpp | 2760 const auto sparse_grad = p_node->Input(3).toBool(); in __anon5d9c3eb910802() local
|