| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | VariableTypeManual.cpp | 261 const Tensor& resize_as_( in resize_as_() function
273 at::redispatch::resize_as_( in resize_as_()
352 "resize_as_", in TORCH_LIBRARY_IMPL()
354 DispatchKey::Autograd, TORCH_FN(VariableType::resize_as_))); in TORCH_LIBRARY_IMPL()
422 static const Tensor& resize_as_( in resize_as_() function
433 at::redispatch::resize_as_( in resize_as_()
548 "resize_as_", in TORCH_LIBRARY_IMPL()
550 DispatchKey::ADInplaceOrView, TORCH_FN(ADInplaceOrView::resize_as_))); in TORCH_LIBRARY_IMPL()
|
| H A D | TraceTypeManual.cpp | 69 const Tensor& resize_as_( in resize_as_() function
74 jit::tracer::warn("resize_as_", jit::tracer::WARN_RESIZE); in resize_as_()
80 self.resize_as_(the_template, optional_memory_format); in resize_as_()
133 m.impl("resize_as_", resize_as_); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | Pad.mm | 202 output.resize_as_(input);
333 grad_input.resize_as_(input).zero_();
356 grad_input.resize_as_(input).zero_();
380 grad_input.resize_as_(input).zero_();
404 grad_input.resize_as_(input).zero_();
427 grad_input.resize_as_(input).zero_();
452 grad_input.resize_as_(input).zero_();
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensorMath.cpp | 107 r.resize_as_(t); in mul_out_sparse_zerodim()
121 r.resize_as_(t); in mul_out_sparse_zerodim()
123 indices.resize_as_(t._indices()); in mul_out_sparse_zerodim()
175 r.resize_as_(t); in pow_out_sparse_scalar()
177 indices.resize_as_(t._indices()); in pow_out_sparse_scalar()
201 tensor._values().resize_as_(coalesced._values()); in coalesce_()
202 tensor._indices().resize_as_(coalesced._indices()); in coalesce_()
244 r.resize_as_(t_tmp); in div_out_sparse_zerodim()
246 indices.resize_as_(t_tmp._indices()); in div_out_sparse_zerodim()
323 result.resize_as_(dividend_tmp); in floor_divide_out_sparse_zerodim()
[all …]
|
| H A D | SoftMax.cpp | 303 out_values.resize_as_(values); in cpu_sparse_coo_softmax()
304 out_indices.resize_as_(indices); in cpu_sparse_coo_softmax()
422 values.resize_as_(out_values); in cpu_sparse_coo_softmax_backward()
424 indices.resize_as_(out_indices); in cpu_sparse_coo_softmax_backward()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | _functions.cpp | 30 output.resize_as_(input); in forward()
31 ctx->saved_data["scale"].toTensor().resize_as_(input); in forward()
110 grad_input.resize_as_(input); in backward()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | SparseCsrTensorImpl.cpp | 177 crow_indices_.resize_as_(compressed_indices); in resize_as_sparse_compressed_tensor_()
180 col_indices_.resize_as_(plain_indices); in resize_as_sparse_compressed_tensor_()
189 values_.resize_as_(src.values()); in resize_as_sparse_compressed_tensor_()
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | _functions.py | 233 output.resize_as_(input)
234 ctx.scale.resize_as_(input)
287 grad_input.resize_as_(input)
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Resize.cpp | 114 // Strides of the output tensor of `resize_as_` operator is defined by input
136 const Tensor& resize_as_( in resize_as_() function
143 "Unsupported memory format for sparse tensor resize_as_ :", in resize_as_()
|
| H A D | LossMultiLabelMargin.cpp | 133 is_target.resize_as_(target); in multilabel_margin_loss_forward_out_cpu_template()
243 grad_input.resize_as_(input); in multilabel_margin_loss_backward_out_cpu_template()
|
| H A D | ReplicationPadding.cpp | 302 gradInput.resize_as_(input, input.suggest_memory_format()); in replication_pad2d_backward_out_cpu()
334 gradInput.resize_as_(input, input.suggest_memory_format()); in replication_pad3d_backward_out_cpu()
|
| H A D | DistributionTemplates.h | 382 // result.resize_as_(self) requires self to have same dtype as result, so we in bernoulli_out_impl()
384 // TODO: Fix resize_as_. See pytorch/pytorch#11665. in bernoulli_out_impl()
|
| H A D | ResizeCommon.h | 39 "Cannot resize named tensor with resize_ or resize_as_ (tried to resize " in resize_named_tensor_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SoftMax.cu | 398 out_values.resize_as_(values); in cuda_sparse_coo_softmax()
399 out_indices.resize_as_(indices); in cuda_sparse_coo_softmax()
476 values.resize_as_(out_values); in cuda_sparse_coo_softmax_backward()
478 indices.resize_as_(out_indices); in cuda_sparse_coo_softmax_backward()
|
| H A D | SparseCUDATensorMath.cu | 283 r_.resize_as_(dense); in add_out_dense_sparse_cuda()
300 r.resize_as_(dense); in add_out_dense_sparse_cuda()
425 promoted.resize_as_(src); in add_out_sparse_cuda()
432 r_.resize_as_(src); in add_out_sparse_cuda()
492 r_.resize_as_(t_); in mul_out_sparse_cuda()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | scalar_tensor_test.cpp | 178 // resize_as_ in test()
182 lhs.resize_as_(rhs); in test()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | MultiLabelMarginCriterion.cu | 217 is_target_.resize_as_(target); in multilabel_margin_loss_forward_out_cuda_template()
323 grad_input.resize_as_(input_); in multilabel_margin_loss_backward_cuda_out_template()
|
| H A D | MaxUnpooling.cu | 461 grad_input.resize_as_(self); in max_unpooling2d_backward_out_cuda()
546 grad_input.resize_as_(self); in max_unpooling3d_backward_out_cuda()
|
| H A D | DilatedMaxPool3d.cu | 480 gradInput.resize_as_(input); in max_pool3d_with_indices_backward_out_cuda_template()
482 gradInput.resize_as_(_input, at::MemoryFormat::ChannelsLast3d); in max_pool3d_with_indices_backward_out_cuda_template()
|
| H A D | ReplicationPadding.cu | 265 gradInput.resize_as_(input); in replication_pad2d_backward_out_cuda_template()
380 gradInput.resize_as_(input); in replication_pad3d_backward_out_cuda_template()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_xpu.py | 46 "resize_as_",
60 "resize_as_", # Skipped by CPU
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/ |
| H A D | ts_native_functions.cpp | 83 dst.resize_as_(typed_tensor).copy_(typed_tensor); in _copy_from()
128 dst.resize_as_(typed_tensor).copy_(typed_tensor); in _copy_from_and_resize()
|
| /aosp_15_r20/external/pytorch/torch/optim/ |
| H A D | _functional.py | 59 return constructor().resize_as_(grad)
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | fake_impls.py | 249 @register_op_impl(aten.resize_as_.default)
250 def resize_as_(fake_mode, func, *args, **kwargs): function
|
| /aosp_15_r20/external/pytorch/torchgen/aoti/ |
| H A D | fallback_ops.py | 114 "aten.resize_as_.default",
|