- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetTFOpName (0.25 sec)
-
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) -
tensorflow/compiler/mlir/tfr/utils/utils.cc
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) { if (compose_func_name[i] >= 'A' && compose_func_name[i] <= 'Z') {
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/raise_to_tf.cc
const llvm::StringMap<Attribute>& derived_attrs) const { // Create the new op Location loc = call_op.getLoc(); rewriter.setInsertionPointAfter(call_op); std::string tf_op_name = GetTFOpName(call_op.getCallee()); OperationState new_state(loc, tf_op_name, inputs, output_types, attr_list); Operation* new_op = rewriter.create(new_state); if (materialize_derived_attrs_) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0)