Searched refs:get_nested_tensor_impl_or_null (Results 1 – 3 of 3) sorted by relevance
77 auto* nt_self = get_nested_tensor_impl_or_null(self); in NestedTensor_matmul()97 const auto* nt_mat1 = get_nested_tensor_impl_or_null(mat1); in NestedTensor_times_Tensor_plus_Tensor_addmm()143 const auto* query_nt = get_nested_tensor_impl_or_null(query); in NestedTensor_softmax_dropout()
333 if (get_nested_tensor_impl_or_null(tensor) == nullptr) { in get_nested_tensor_structure()
210 inline NestedTensorImpl* get_nested_tensor_impl_or_null( in get_nested_tensor_impl_or_null() function