Searched refs:ops_and_kernels (Results 1 – 2 of 2) sorted by relevance
109 ops_and_kernels = selective_registration_header_lib.get_ops_and_kernels(124 ops_and_kernels)128 ops_and_kernels = selective_registration_header_lib.get_ops_and_kernels(141 ops_and_kernels)148 ops_and_kernels = selective_registration_header_lib.get_ops_and_kernels(153 ], ops_and_kernels)157 ops_and_kernels = selective_registration_header_lib.get_ops_and_kernels(161 ], ops_and_kernels)166 ops_and_kernels = selective_registration_header_lib.get_ops_and_kernels(170 ], ops_and_kernels)[all …]
110 def get_header_from_ops_and_kernels(ops_and_kernels, argument122 ops = set(op for op, _ in ops_and_kernels)166 for _, kernel_class in ops_and_kernels:211 ops_and_kernels = get_ops_and_kernels(proto_fileformat, graphs, default_ops)212 if not ops_and_kernels:216 return get_header_from_ops_and_kernels(ops_and_kernels, default_ops == 'all')