Searched refs:areAnyArgumentsTensorList (Results 1 – 2 of 2) sorted by relevance
65 static bool areAnyArgumentsTensorList(const at::FunctionSchema& schema) { in areAnyArgumentsTensorList() function265 TORCH_CHECK(areAllReturnsTensors(schema) && !areAnyArgumentsTensorList(schema), in batchedTensorForLoopFallback()405 TORCH_CHECK(areAllReturnsTensors(schema) && !areAnyArgumentsTensorList(schema), in batchedNestedTensorForLoopFallback()
35 static bool areAnyArgumentsTensorList(const FunctionSchema& schema) { in areAnyArgumentsTensorList() function261 TORCH_CHECK(areAllReturnsTensors(schema) && !areAnyArgumentsTensorList(schema), in batchedTensorForLoopFallback()