Searched refs:builtin_op_code (Results 1 – 4 of 4) 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 …]
97 const BuiltinOperator builtin_op_code = GetBuiltinCode(op_code); in GetWeightInputIndices() local98 if (builtin_op_code == BuiltinOperator_CUSTOM) { in GetWeightInputIndices()104 } else if (builtin_op_code == BuiltinOperator_CONV_2D || in GetWeightInputIndices()105 builtin_op_code == BuiltinOperator_DEPTHWISE_CONV_2D || in GetWeightInputIndices()106 builtin_op_code == BuiltinOperator_FULLY_CONNECTED || in GetWeightInputIndices()107 builtin_op_code == BuiltinOperator_BATCH_MATMUL || in GetWeightInputIndices()108 builtin_op_code == BuiltinOperator_EMBEDDING_LOOKUP || in GetWeightInputIndices()109 builtin_op_code == BuiltinOperator_TRANSPOSE_CONV) { in GetWeightInputIndices()111 } else if (builtin_op_code == BuiltinOperator_SVDF) { in GetWeightInputIndices()114 } else if (builtin_op_code == BuiltinOperator_LSTM || in GetWeightInputIndices()[all …]
110 return logging_op_resolver_->GetWrappedKernelInvoke(op_info.builtin_op_code, in GetKernelInvoke()200 int builtin_op_code) { in GetLoggingEvalFunc() argument201 switch (builtin_op_code) { in GetLoggingEvalFunc()237 auto builtin_op_code = calibrator->GetOpInfo(node).builtin_op_code; in LoggingEval() local239 GetLoggingEvalFunc(context, node, builtin_op_code); in LoggingEval()417 op_info.builtin_op_code = GetBuiltinCode(operator_code); in BuildLoggingInterpreter()426 } else if (HasInputs(op_info.builtin_op_code)) { in BuildLoggingInterpreter()433 } else if (HasOutputs(op_info.builtin_op_code)) { in BuildLoggingInterpreter()450 {op_info.builtin_op_code, operator_code->version()}); in BuildLoggingInterpreter()
61 BuiltinOperator builtin_op_code; member