Searched defs:op_key (Results 1 – 4 of 4) sorted by relevance
46 auto op_key = std::make_pair(op, version); in AddBuiltin() local65 auto op_key = std::make_pair(name, version); in AddCustom() local
194 def _get_op_from_collection(meta_graph_def, op_key): argument
157 tflite::details::OperatorKey op_key(op_signature, ops_by_type, in CountOperatorsByType() local
407 RngOutput PhiloxRngBit32(XlaOp op_key, XlaOp initial_state, in PhiloxRngBit32()434 RngOutput PhiloxRngBit64(XlaOp op_key, XlaOp initial_state, in PhiloxRngBit64()