Home
last modified time | relevance | path

Searched refs:resize_outputs (Results 1 – 25 of 34) sorted by relevance

12

/external/pytorch/aten/src/ATen/native/quantized/
DTensorAdvancedIndexing.cpp23 config.resize_outputs(false); in make_index_put_iterator()
49 .resize_outputs(false) in masked_fill_impl_quantized_cpu()
101 .resize_outputs(false) in masked_fill_impl_quantized_cuda()
/external/pytorch/aten/src/ATen/native/cuda/
DScatterGatherKernel.cu189 .resize_outputs(false) in operator ()()
247 .resize_outputs(false) in operator ()()
306 .resize_outputs(false) in operator ()()
402 .resize_outputs(false) in operator ()()
445 .resize_outputs(false) in operator ()()
Dgroup_norm_kernel.cu497 .resize_outputs(false) in GroupNorm1dForward()
512 .resize_outputs(false) in GroupNorm1dForward()
525 .resize_outputs(false) in GroupNorm1dForward()
539 .resize_outputs(false) in GroupNorm1dForward()
591 .resize_outputs(false) in GroupNormKernelImplInternal()
623 .resize_outputs(false) in GroupNormKernelImplInternal()
720 .resize_outputs(false) in GroupNorm1dBackward()
740 .resize_outputs(false) in GroupNorm1dBackward()
896 .resize_outputs(false) in GroupNormBackwardKernelImplInternal()
912 .resize_outputs(false) in GroupNormBackwardKernelImplInternal()
DActivation.cpp49 .resize_outputs(false) in glu_backward_cuda_out()
DCrossKernel.cu73 .resize_outputs(false) in cross_impl()
DIndexKernel.cu402 .resize_outputs(false) in launch_masked_scatter_kernel()
/external/executorch/backends/xnnpack/runtime/
DXNNExecutor.h79 ET_NODISCARD executorch::runtime::Error resize_outputs(
DXNNPACKBackend.cpp128 err = executor->resize_outputs(args); in execute()
DXNNExecutor.cpp188 ET_NODISCARD Error XNNExecutor::resize_outputs(EValue** args) const { in resize_outputs() function in executorch::backends::xnnpack::delegate::XNNExecutor
/external/pytorch/aten/src/ATen/native/
DUnfoldBackward.h101 .resize_outputs(false) in _make_unfold_backward_iter_over_grad_out()
DFunctionOfAMatrixUtils.cpp98 .resize_outputs(false) in _compute_linear_combination_out()
DCopy.cpp275 .resize_outputs(false) in copy_impl()
377 .resize_outputs(false) in copy_ignoring_overlaps()
DFill.cpp33 .resize_outputs(false) in fill_out()
DEmbedding.cpp128 .resize_outputs(false) in embedding_dense_backward_cpu()
DTensorAdvancedIndexing.cpp614 config.resize_outputs(false); in make_index_put_iterator()
925 .resize_outputs(false) in TORCH_IMPL_FUNC()
1357 .resize_outputs(false) in index_select_out_cpu_()
1539 .resize_outputs(false) in index_fill_()
2000 .resize_outputs(false) in masked_fill_impl_cpu()
2090 .resize_outputs(false) in masked_select_out_impl_cpu()
2114 .resize_outputs(false) in masked_select_out_impl_cpu()
2569 .resize_outputs(false) in masked_scatter__cpu()
DNormalization.cpp252 .resize_outputs(false) in batch_norm_cpu_update_stats_template()
369 .resize_outputs(false) in batch_norm_backward_cpu_template()
380 .resize_outputs(false) in batch_norm_backward_cpu_template()
389 .resize_outputs(false) in batch_norm_backward_cpu_template()
DSorting.cpp448 .resize_outputs(false) in kthvalue_out_impl_cpu()
545 .resize_outputs(false) in median_with_indices_impl()
/external/pytorch/aten/src/ATen/native/cpu/
DSortingKernel.cpp36 .resize_outputs(false) in _dim_apply()
238 .resize_outputs(false) in topk_kernel()
DScatterGatherKernel.cpp185 .resize_outputs(false) in operator ()()
272 .resize_outputs(false) in operator ()()
368 .resize_outputs(false) in operator ()()
463 .resize_outputs(false) in operator ()()
559 .resize_outputs(false) in operator ()()
DTensorCompareKernel.cpp58 .resize_outputs(false) in compare_base_kernel_core()
DReduceOpsKernel.cpp51 .resize_outputs(false) in cpu_cum_base_kernel()
/external/pytorch/aten/src/ATen/
DTensorIterator.h942 TensorIteratorConfig& resize_outputs(bool resize_outputs) { in resize_outputs() argument
943 resize_outputs_ = resize_outputs; in resize_outputs()
DTensorIterator.cpp1105 .resize_outputs(false)
1125 .resize_outputs(false) in reduce_op()
1149 .resize_outputs(false) in reduce_op()
/external/pytorch/aten/src/ATen/native/mps/operations/
DCrossKernel.mm90 .resize_outputs(false)
/external/pytorch/aten/src/ATen/native/sparse/
DSparseBinaryOpIntersectionCommon.h114 .resize_outputs(false) in make_value_selection_intersection_iter()

12