Home
last modified time | relevance | path

Searched refs:check_for_empty (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/
H A Dtransformer.cpp97 … const Tensor& check_for_empty = src.is_nested() ? get_nested_tensor_impl(src)->get_buffer() : src; in transformer_encoder_layer_forward() local
98 if (check_for_empty.numel() == 0) { in transformer_encoder_layer_forward()
100 …? at::detail::make_tensor<NestedTensorImpl>(check_for_empty, get_nested_tensor_impl(src)->get_nest… in transformer_encoder_layer_forward()