Searched refs:kernel_class (Results 1 – 2 of 2) sorted by relevance
67 kernel_class = _pywrap_kernel_registry.TryFindKernelClass(70 if kernel_class or op in OPS_WITHOUT_KERNEL_ALLOWLIST:71 op_and_kernel = (op, str(kernel_class.decode('utf-8'))72 if kernel_class else None)166 for _, kernel_class in ops_and_kernels:167 if kernel_class is None:169 line += '"%s",\n' % kernel_class
121 std::string kernel_class = FindTensorflowKernelClass(&nodedef); in AddFlexOpsFromModel() local122 flex_ops->insert({tf_op_name, kernel_class}); in AddFlexOpsFromModel()