- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for compose_func_name (0.17 sec)
-
tensorflow/compiler/mlir/tfr/utils/utils.cc
compose_func_name.push_back('_'); compose_func_name.push_back(tf_op_name[i] + 'a' - 'A'); } else { compose_func_name.push_back(tf_op_name[i]); } } return compose_func_name; } std::string GetTFOpName(StringRef compose_func_name) { std::string tf_op_name; bool after_underscore = false; for (int i = 0; i < compose_func_name.size(); ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
// bridge. if (op->isRegistered()) { return WalkResult::advance(); } // Find out the compose function auto compose_func_name = GetComposeFuncName(op->getName().getStringRef()); auto compose_func = table.lookup<TFRFuncOp>(compose_func_name); if (!compose_func || compose_func.isExternal()) { // There are no decomposition methods defined for this op, skip. return WalkResult::advance();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/utils/utils.h
// This is a hardcoded rule for mapping a TFR function op name to the // corresponding TF opname. Examples: // tf__pack -> tf.Pack // tf__concat_v2 => tf.ConcatV2 std::string GetTFOpName(StringRef compose_func_name); // Validate the attributes of 'src' is either contained in the registered // attribute sets or in the allowed list. LogicalResult ValidateAttrs(Operation* src, const StringSet<>& registered);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 2.6K bytes - Viewed (0)