Searched defs:nested_tensor (Results 1 – 3 of 3) sorted by relevance
194 def nested_tensor(tensor_list, *, dtype=None, layout=None, device=None, requires_grad=False, pin_me… function
16 const Tensor& nested_tensor, in check_nested_tensor_matrix_constraints()
222 for (const auto& nested_tensor : tensor.variant_tensors) { in GetTensorType() local