- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for CompositeOpExpansion (0.17 sec)
-
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) -
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)