Searched defs:op_set (Results 1 – 5 of 5) sorted by relevance
47 const OpSet& op_set) { in InsertQuantizedFunctionsPass()101 QuantizationMethod quantization_method, OpSet op_set) { in GetFunctionLibrary()178 QuantizationMethod quantization_method, const OpSet& op_set) { in CreateInsertQuantizedFunctionsPass()
126 OpSet op_set = 2; field
407 Status CheckInternalDataDependency(const std::set<string>& op_set, in CheckInternalDataDependency()423 void ClearInternalControlInputs(const std::set<string>& op_set, in ClearInternalControlInputs()
103 absl::flat_hash_set<HloInstruction*> op_set; in CountNonLeafOps() local
236 def register_ops_with_aoti_compile(ns, op_set, dispatch_key, torch_compile_op_lib_impl): argument