Searched refs:unsafe_replace_operand (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Normalization.cpp | 262 iter.unsafe_replace_operand(0, in_data + channel_stride * f); in batch_norm_cpu_update_stats_template()
420 reduce_iter_local.unsafe_replace_operand( in batch_norm_backward_cpu_template()
422 reduce_iter_local.unsafe_replace_operand( in batch_norm_backward_cpu_template()
439 unary_iter_local.unsafe_replace_operand( in batch_norm_backward_cpu_template()
441 unary_iter_local.unsafe_replace_operand( in batch_norm_backward_cpu_template()
451 binary_iter_local.unsafe_replace_operand(0, gI_data); in batch_norm_backward_cpu_template()
452 binary_iter_local.unsafe_replace_operand(1, gI_data); in batch_norm_backward_cpu_template()
453 binary_iter_local.unsafe_replace_operand( in batch_norm_backward_cpu_template()
465 unary_iter_local.unsafe_replace_operand( in batch_norm_backward_cpu_template()
467 unary_iter_local.unsafe_replace_operand( in batch_norm_backward_cpu_template()
|
| H A D | Embedding.cpp | 165 iter.unsafe_replace_operand(0, gW_data + k * gW_stride); in embedding_dense_backward_cpu()
166 iter.unsafe_replace_operand(1, gW_data + k * gW_stride); in embedding_dense_backward_cpu()
167 iter.unsafe_replace_operand(2, const_cast<char*>(gO_data + i * gO_stride)); in embedding_dense_backward_cpu()
|
| H A D | LossCTC.cpp | 303 fill_iter_local.unsafe_replace_operand(0, grad_a.data()); in ctc_loss_backward_cpu_template()
320 fill_log_beta_1d_iter_local.unsafe_replace_operand( in ctc_loss_backward_cpu_template()
397 fill_1d_iter_local.unsafe_replace_operand(0, grad_a[l].data()); in ctc_loss_backward_cpu_template()
|
| H A D | TensorAdvancedIndexing.cpp | 1028 iter.unsafe_replace_operand(0, self_data); in TORCH_IMPL_FUNC()
1029 iter.unsafe_replace_operand(1, self_data); in TORCH_IMPL_FUNC()
1030 iter.unsafe_replace_operand(2, const_cast<char*>(source_data)); in TORCH_IMPL_FUNC()
1125 iter.unsafe_replace_operand(0, self_data); in index_reduce_func_impl()
1126 iter.unsafe_replace_operand(1, self_data); in index_reduce_func_impl()
1127 iter.unsafe_replace_operand(2, const_cast<char*>(source_data)); in index_reduce_func_impl()
1378 sub_iter.unsafe_replace_operand(0, result_data); in index_select_out_cpu_()
1379 sub_iter.unsafe_replace_operand(1, const_cast<char*>(self_data)); in index_select_out_cpu_()
|
| H A D | TensorShape.cpp | 634 iter.unsafe_replace_operand(0, result_data); in TORCH_IMPL_FUNC()
635 iter.unsafe_replace_operand(1, const_cast<char*>(source_data)); in TORCH_IMPL_FUNC()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | TensorIterator.h | 374 void unsafe_replace_operand(int64_t arg, void* data);
|
| H A D | TensorIterator.cpp | 854 void TensorIteratorBase::unsafe_replace_operand(int64_t arg, void* data) { in unsafe_replace_operand() function in at::TensorIteratorBase
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensorMath.cpp | 1869 copy_iter_local.unsafe_replace_operand(0, gIv_data + i * gIv_stride); in _sparse_sum_backward_cpu()
1870 copy_iter_local.unsafe_replace_operand(1, gOv_data + m * gOv_stride); in _sparse_sum_backward_cpu()
|