Home
last modified time | relevance | path

Searched refs:_log_softmax_backward_data (Results 1 – 25 of 123) sorted by relevance

12345

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSoftMax.cpp433 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 DBatchRulesReduceOps.cpp314 …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 DSoftMax.cu492 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 Dconvmixer_768_32_training.txt3 Operator: aten._log_softmax_backward_data.default
H A Dresmlp_12_224_training.txt3 Operator: aten._log_softmax_backward_data.default
H A Dmixer_b16_224_training.txt3 Operator: aten._log_softmax_backward_data.default
H A Dgmlp_s16_224_training.txt3 Operator: aten._log_softmax_backward_data.default
H A Dvit_base_patch16_224_training.txt3 Operator: aten._log_softmax_backward_data.default
H A Ddeit_base_distilled_patch16_224_training.txt3 Operator: aten._log_softmax_backward_data.default
/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/torchbench_train/
H A Dpytorch_struct_training.txt5 Operator: aten._log_softmax_backward_data.default
/aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/hf_train/
H A DPegasusForCausalLM_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DMBartForCausalLM_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DPLBartForCausalLM_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DBartForCausalLM_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DBlenderbotSmallForCausalLM_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DTrOCRForCausalLM_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DBertForMaskedLM_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DGoogleFnet_training.txt6 Operator: aten._log_softmax_backward_data.default
H A DBlenderbotSmallForConditionalGeneration_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DDistilBertForMaskedLM_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DPegasusForConditionalGeneration_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DSpeech2Text2ForCausalLM_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DDebertaV2ForMaskedLM_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DBartForConditionalGeneration_training.txt3 Operator: aten._log_softmax_backward_data.default
H A DBertForQuestionAnswering_training.txt3 Operator: aten._log_softmax_backward_data.default

12345