Searched refs:builtin_op_code (Results 1 – 3 of 3) sorted by relevance
83 const BuiltinOperator builtin_op_code = GetBuiltinCode(op_code); in GetWeightInputIndices() local84 if (builtin_op_code == BuiltinOperator_CUSTOM) { in GetWeightInputIndices()90 } else if (builtin_op_code == BuiltinOperator_CONV_2D || in GetWeightInputIndices()91 builtin_op_code == BuiltinOperator_DEPTHWISE_CONV_2D || in GetWeightInputIndices()92 builtin_op_code == BuiltinOperator_FULLY_CONNECTED || in GetWeightInputIndices()93 builtin_op_code == BuiltinOperator_BATCH_MATMUL || in GetWeightInputIndices()94 builtin_op_code == BuiltinOperator_EMBEDDING_LOOKUP || in GetWeightInputIndices()95 builtin_op_code == BuiltinOperator_TRANSPOSE_CONV) { in GetWeightInputIndices()97 } else if (builtin_op_code == BuiltinOperator_SVDF) { in GetWeightInputIndices()100 } else if (builtin_op_code == BuiltinOperator_LSTM || in GetWeightInputIndices()[all …]
108 return logging_op_resolver_->GetWrappedKernelInvoke(op_info.builtin_op_code, in GetKernelInvoke()198 int builtin_op_code) { in GetLoggingEvalFunc() argument199 switch (builtin_op_code) { in GetLoggingEvalFunc()235 auto builtin_op_code = calibrator->GetOpInfo(node).builtin_op_code; in LoggingEval() local237 GetLoggingEvalFunc(context, node, builtin_op_code); in LoggingEval()411 op_info.builtin_op_code = GetBuiltinCode(operator_code); in BuildLoggingInterpreter()420 } else if (HasInputs(op_info.builtin_op_code)) { in BuildLoggingInterpreter()428 } else if (HasOutputs(op_info.builtin_op_code)) { in BuildLoggingInterpreter()446 {op_info.builtin_op_code, operator_code->version()}); in BuildLoggingInterpreter()
60 BuiltinOperator builtin_op_code; member