Searched defs:use_nested_tensor (Results 1 – 1 of 1) sorted by relevance
69 const bool use_nested_tensor) { in norm()104 const bool use_nested_tensor = src.is_nested(); in transformer_encoder_layer_forward() local