- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for _XlaCompiledKernel (0.14 sec)
-
tensorflow/compiler/jit/build_xla_ops_pass.h
#include "absl/types/optional.h" #include "tensorflow/core/common_runtime/optimization_registry.h" #include "tensorflow/core/lib/core/status.h" namespace tensorflow { // Replaces TF function calls marked with `_XlaCompiledKernel` with _XlaCompile // and _XlaRun nodes (which compile and launch, respectively, the corresponding // HLO module). class BuildXlaOpsPass : public GraphOptimizationPass { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 03 16:15:20 UTC 2022 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.h
// (derived from kXlaClusterAttr=ID (kXlaClusterAttr) attribute), puts them into // a TF function, and replaces the subgraph in the main graph with a call to // that TF function annotated with kXlaCompiledKernelAttr (_XlaCompiledKernel). class EncapsulateSubgraphsPass : public GraphOptimizationPass { public: Status Run(const GraphOptimizationPassOptions& options) override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 12 03:59:36 UTC 2022 - 4.9K bytes - Viewed (0)