- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetMlirOpNameFromOpCode (0.9 sec)
-
tensorflow/compiler/mlir/lite/flatbuffer_operator.h
bool IsStablehloOp(const tflite::OperatorCodeT &op_code); // Returns the MLIR op name for the flatbuffer operator corresponding to // `op_code`. std::string GetMlirOpNameFromOpCode(const ::tflite::OperatorCodeT &op_code); // Returns the builtin op code for the given MLIR operation on success; emits // error and returns std::nullopt on failure.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 21:00:09 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
llvm::StringRef op_name( tflite::EnumNameBuiltinOperator(tflite::GetBuiltinCode(&op_code))); return op_name.starts_with("STABLEHLO_"); } std::string mlir::GetMlirOpNameFromOpCode( const tflite::OperatorCodeT& op_code) { auto builtin_code = tflite::GetBuiltinCode(&op_code); if (builtin_code == tflite::BuiltinOperator_IF) { return std::string("tf.If"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
const tflite::OperatorCodeT& op_code) { if (IsBasicLSTMOp(op.builtin_options)) { return std::string("tfl.basic_lstm"); } return mlir::GetMlirOpNameFromOpCode(op_code); } StatusOr<Operation*> BuildExternalConstOp(const tflite::TensorT& tensor, int32_t buffer_index,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0)