Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetOperationNodeDef (0.44 sec)

  1. tensorflow/compiler/mlir/tensorflow/utils/export_utils.h

    // Converts an MLIR operation to TensorFlow NodeDef with given node name. This
    // name should be unique to the graph it is being inserted into.
    absl::StatusOr<std::unique_ptr<NodeDef>> GetOperationNodeDef(
        mlir::Operation* inst, llvm::StringRef name);
    
    // Converts MLIR attributes with values to their tensorflow equivalent.
    // "name" and "device" attributes are ignored by default. Use attrs_to_ignore to
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 3.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc

    }
    
    absl::StatusOr<std::unique_ptr<NodeDef>> ConvertTFDialectOpToNodeDef(
        mlir::Operation* inst, llvm::StringRef name,
        bool ignore_unregistered_attrs) {
      TF_ASSIGN_OR_RETURN(auto node_def, GetOperationNodeDef(inst, name));
      TF_ASSIGN_OR_RETURN(auto op_name,
                          GetTensorFlowOpName(inst->getName().getStringRef()));
      const tensorflow::OpRegistrationData* op_reg_data =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

      // NextIteration sink ends with ".Sink".
      if (!op_name.consume_back(".sink")) op_name.consume_back(".Sink");
      return op_name;
    }
    
    absl::StatusOr<std::unique_ptr<NodeDef>> GetOperationNodeDef(
        mlir::Operation* inst, llvm::StringRef name) {
      auto node_def = std::make_unique<NodeDef>();
      // Note: we do not use NodeBuilder or NodeDefBuilder as that would require
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
Back to top