| /external/pytorch/aten/src/ATen/native/quantized/ |
| D | TensorAdvancedIndexing.cpp | 40 "Use of masked_fill_ on expanded tensors is deprecated. " in masked_fill_impl_quantized_cpu()
61 auto maybe_outnames = namedinference::broadcast_to_outnames(self, mask, "masked_fill_"); in masked_fill__quantized_cpu()
70 auto maybe_outnames = namedinference::broadcast_to_outnames(self, mask, "masked_fill_"); in masked_fill__quantized_cpu()
71 …TORCH_CHECK(value.dim() == 0, "masked_fill_ only supports a 0-dimensional value tensor, but got te… in masked_fill__quantized_cpu()
86 auto maybe_outnames = namedinference::broadcast_to_outnames(self, mask, "masked_fill_"); in masked_fill_impl_quantized_cuda()
90 "Use of masked_fill_ on expanded tensors is deprecated. " in masked_fill_impl_quantized_cuda()
96 c10::MaybeOwned<Tensor> b_mask = expand_inplace(self, mask, "masked_fill_"); in masked_fill_impl_quantized_cuda()
113 TORCH_CHECK(!self.device().is_cpu(), "masked_fill_: Expected inputs to be on same device") in masked_fill__quantized_cuda()
118 …TORCH_CHECK(value.dim() == 0, "masked_fill_ only supports a 0-dimensional value tensor, but got te… in masked_fill__quantized_cuda()
120 TORCH_CHECK(!self.device().is_cpu(), "masked_fill_: Expected inputs to be on same device") in masked_fill__quantized_cuda()
[all …]
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Indexing.cu | 1193 counts.masked_fill_(counts == 0, 1); in TORCH_IMPL_FUNC()
1515 kBool, kHalf, kBFloat16, kComplexHalf, iter.common_dtype(), "masked_fill_", [&]() { in masked_fill_kernel()
1542 iter.common_dtype(), "masked_fill_", [&]() { in masked_fill_kernel_quantized()
1559 auto maybe_outnames = namedinference::broadcast_to_outnames(self, mask, "masked_fill_"); in masked_fill__cuda()
1562 "Use of masked_fill_ on expanded tensors is deprecated. " in masked_fill__cuda()
1568 c10::MaybeOwned<Tensor> b_mask = expand_inplace(self, mask, "masked_fill_"); in masked_fill__cuda()
1585 …TORCH_CHECK(value.dim() == 0, "masked_fill_ only supports a 0-dimensional value tensor, but got te… in masked_fill__cuda()
1591 TORCH_CHECK(!self.device().is_cpu(), "masked_fill_: Expected inputs to be on same device") in masked_fill__cuda()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorAdvancedIndexing.cpp | 800 return self.masked_fill_(std::get<1>(masked_fill_dispatch), value.item()); in _index_put_impl_()
1149 counts.masked_fill_(counts == 0, 1); in index_reduce_func_impl()
1204 counts.masked_fill_(counts == 0, 1); in index_reduce_func_impl()
1948 count.masked_fill_(count == 0, 1); in TORCH_IMPL_FUNC()
1986 …TORCH_CHECK(mask.dtype() == ScalarType::Bool, "masked_fill_ only supports boolean masks, but got m… in masked_fill_impl_cpu()
1991 "Use of masked_fill_ on expanded tensors is deprecated. " in masked_fill_impl_cpu()
2010 auto maybe_outnames = namedinference::broadcast_to_outnames(self, mask, "masked_fill_"); in masked_fill__cpu()
2018 auto maybe_outnames = namedinference::broadcast_to_outnames(self, mask, "masked_fill_"); in masked_fill__cpu()
2019 …TORCH_CHECK(value.dim() == 0, "masked_fill_ only supports a 0-dimensional value tensor, but got te… in masked_fill__cpu()
2034 result.masked_fill_(mask, source); in masked_fill()
[all …]
|
| D | LossNLL.cpp | 585 smooth_loss.masked_fill_(ignore_mask, 0.0); in cross_entropy_loss_label_smoothing()
597 tgt_weights.masked_fill_(ignore_mask.flatten(), 0).sum(); in cross_entropy_loss_label_smoothing()
|
| /external/pytorch/test/ |
| D | test_maskedtensor.py | 38 a = mt_result_data.detach().masked_fill_(~mask, 0)
39 b = t_result.detach().masked_fill_(~mask, 0)
63 a = mt_data1.detach().masked_fill_(~mask, 0)
64 b = mt_data2.detach().masked_fill_(~mask, 0)
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | remove_inplace_ops.cpp | 11 {aten::masked_fill_, aten::masked_fill},
|
| /external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/torchbench_train/ |
| D | tts_angular_training.txt | 29 Operator: aten.masked_fill_.Scalar
|
| D | hf_Bart_training.txt | 53 Operator: aten.masked_fill_.Scalar
|
| /external/pytorch/benchmarks/dynamo/microbenchmarks/ |
| D | operatorbench.py | 120 "masked_fill_.Scalar",
|
| /external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/hf_train/ |
| D | PegasusForCausalLM_training.txt | 47 Operator: aten.masked_fill_.Scalar
|
| D | BartForCausalLM_training.txt | 48 Operator: aten.masked_fill_.Scalar
|
| D | PLBartForCausalLM_training.txt | 48 Operator: aten.masked_fill_.Scalar
|
| D | MBartForCausalLM_training.txt | 48 Operator: aten.masked_fill_.Scalar
|
| D | BlenderbotSmallForCausalLM_training.txt | 48 Operator: aten.masked_fill_.Scalar
|
| D | PegasusForConditionalGeneration_training.txt | 54 Operator: aten.masked_fill_.Scalar
|
| D | TrOCRForCausalLM_training.txt | 48 Operator: aten.masked_fill_.Scalar
|
| D | XGLMForCausalLM_training.txt | 58 Operator: aten.masked_fill_.Scalar
|
| D | DebertaV2ForMaskedLM_training.txt | 54 Operator: aten.masked_fill_.Scalar
|
| D | Speech2Text2ForCausalLM_training.txt | 50 Operator: aten.masked_fill_.Scalar
|
| D | BartForConditionalGeneration_training.txt | 61 Operator: aten.masked_fill_.Scalar
|
| D | M2M100ForConditionalGeneration_training.txt | 56 Operator: aten.masked_fill_.Scalar
|
| D | BlenderbotSmallForConditionalGeneration_training.txt | 55 Operator: aten.masked_fill_.Scalar
|
| D | MBartForConditionalGeneration_training.txt | 62 Operator: aten.masked_fill_.Scalar
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesBinaryOps.cpp | 478 …VMAP_SUPPORT2(masked_fill_, Scalar, SINGLE_ARG(binary_pointwise_inplace_batch_rule<TensorScalarInp… in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/aten/src/ATen/native/mps/operations/ |
| D | Indexing.mm | 663 auto maybe_outnames = namedinference::broadcast_to_outnames(self, mask, "masked_fill_");
665 c10::MaybeOwned<Tensor> b_mask = expand_inplace(self, mask, "masked_fill_");
816 "masked_fill_ only supports a 0-dimensional value tensor, but got tensor "
|