Searched refs:op_key (Results 1 – 3 of 3) sorted by relevance
40 auto op_key = std::make_pair(op, version); in AddBuiltin() local41 builtins_[op_key] = new_registration; in AddBuiltin()53 auto op_key = std::make_pair(name, version); in AddCustom() local54 custom_ops_[op_key] = new_registration; in AddCustom()
163 def _get_op_from_collection(meta_graph_def, op_key): argument164 return _get_main_op_tensor(meta_graph_def, op_key)
1205 op_key = (pooling_type, num_spatial_dims)1206 if op_key not in pooling_ops:1207 raise ValueError("%d-D %s pooling is not supported." % (op_key[1],1208 op_key[0]))1235 result = pooling_ops[op_key](