Searched defs:arg_out (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | SpmmReduceKernel.cpp | 165 const Tensor& arg_out, in spmm_reduce_arg_kernel_impl()
306 auto arg_out = arg_out_.contiguous(); in spmm_reduce_backward_input_arg_kernel_impl() local
389 auto arg_out = arg_out_.contiguous(); in spmm_reduce_backward_other_arg_kernel_impl() local
449 const Tensor& arg_out, in spmm_reduce_arg_kernel()
489 const Tensor& arg_out, in spmm_reduce_backward_input_arg_kernel()
545 const Tensor& arg_out, in spmm_reduce_backward_other_arg_kernel()
|
| /external/bcc/src/cc/ |
| D | bpf_module_rw_engine.cc | 279 Argument *arg_out = &*arg_it; in make_reader() local
336 Argument *arg_out = &*arg_it; in make_writer() local
|
| /external/pytorch/aten/src/ATen/native/sparse/ |
| D | SparseCsrTensorMath.cpp | 1385 auto arg_out = at::empty({0}, col.options()); in _sparse_mm_reduce_impl_sparse_csr_cpu() local
1415 const Tensor& arg_out, in _sparse_mm_reduce_impl_backward_sparse_csr_cpu()
|
| /external/libevent/ |
| D | event.c | 2419 …e_out, evutil_socket_t *fd_out, short *events_out, event_callback_fn *callback_out, void **arg_out) in event_get_assignment()
|