/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
H A D | SoftMax.cpp | 433 auto r = at::cpu::_log_softmax_backward_data( in cpu_sparse_coo_softmax_backward() 446 auto r = at::cpu::_log_softmax_backward_data( in cpu_sparse_coo_softmax_backward()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
H A D | BatchRulesReduceOps.cpp | 314 …return std::make_tuple(at::_log_softmax_backward_data(grad_output_, output_, dim, input_dtype), 0); in _log_softmax_backward_batch_rule() 506 VMAP_SUPPORT(_log_softmax_backward_data, _log_softmax_backward_batch_rule); in TORCH_LIBRARY_IMPL()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
H A D | SoftMax.cu | 492 auto r = at::cuda::_log_softmax_backward_data( in cuda_sparse_coo_softmax_backward() 518 auto r = at::cuda::_log_softmax_backward_data( in cuda_sparse_coo_softmax_backward()
|
/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/timm_train/ |
H A D | convmixer_768_32_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | resmlp_12_224_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | mixer_b16_224_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | gmlp_s16_224_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | vit_base_patch16_224_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | deit_base_distilled_patch16_224_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/torchbench_train/ |
H A D | pytorch_struct_training.txt | 5 Operator: aten._log_softmax_backward_data.default
|
/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/hf_train/ |
H A D | PegasusForCausalLM_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | MBartForCausalLM_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | PLBartForCausalLM_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | BartForCausalLM_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | BlenderbotSmallForCausalLM_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | TrOCRForCausalLM_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | BertForMaskedLM_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | GoogleFnet_training.txt | 6 Operator: aten._log_softmax_backward_data.default
|
H A D | BlenderbotSmallForConditionalGeneration_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | DistilBertForMaskedLM_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | PegasusForConditionalGeneration_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | Speech2Text2ForCausalLM_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | DebertaV2ForMaskedLM_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | BartForConditionalGeneration_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|
H A D | BertForQuestionAnswering_training.txt | 3 Operator: aten._log_softmax_backward_data.default
|