Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for CompositeOpExpansion (0.17 sec)

  1. tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.h

    // to the decomposition library. Currently the decomposition library is loaded
    // each time the pass runs. A special environment variable is set to locate the
    // decomposition library.
    class CompositeOpExpansion : public EagerOpRewrite {
     public:
      CompositeOpExpansion(string name, string file, string line)
          : EagerOpRewrite(name, file, line) {}
    
      Status Run(EagerOperation* orig_op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 1.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.cc

        monitoring::Counter<0>::New("/tensorflow/core/op_expansion/node_counter",
                                    "The number of nodes being op expanded.");
    }  // namespace
    
    namespace tfr {
    
    Status CompositeOpExpansion::Run(EagerOperation* orig_op,
                                     std::unique_ptr<EagerOperation>* out_op) {
      if (!IsEnabled()) return absl::OkStatus();
      // This can be the default cpu device.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Feb 25 16:22:36 UTC 2024
    - 3.8K bytes
    - Viewed (0)
Back to top