- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for kXlaCompileAttr (0.22 sec)
-
tensorflow/compiler/jit/defs.cc
#include "tensorflow/compiler/jit/defs.h" #include <atomic> namespace tensorflow { const char* const kXlaMustCompileAttr = "_XlaMustCompile"; const char* const kXlaCompileAttr = "_XlaCompile"; // User-provided through jit_scope APIs. Effective only when auto_jit is OFF. const char* const kXlaScopeAttr = "_XlaScope"; // Automatically inserted by auto_jit to guide clustering results. Effective
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 07 01:03:32 UTC 2021 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/jit/defs.h
extern const char* const kXlaMustCompileAttr; // "_XlaMustCompile" // Implies auto-clustering: tagged nodes will be clustered and compiled with XLA // on a best-effort basis. extern const char* const kXlaCompileAttr; // "_XlaCompile" // Implies auto-clustering within the given scope. extern const char* const kXlaScopeAttr; // "_XlaScope" extern const char* const kXlaInternalScopeAttr; // "_XlaInternalScope"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 07 01:03:32 UTC 2021 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator.h
public: // Given a NodeDef 'node_def' and the function library runtime 'flr', returns // true if 'node_def' is a call to a compilable function defined in 'flr', // with the kXlaCompileAttr set. bool CanCreateKernel( const FunctionLibraryRuntime& flr, const std::shared_ptr<const NodeProperties>& props) const override;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 26 19:43:17 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass_test.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
bool compile = false; Status status = GetNodeAttr(node->attrs(), kXlaCompileAttr, &compile); if (status.ok()) { if (!compile) { VLOG(2) << "Rejecting " << node->name() << ": kXlaCompileAttr(" << kXlaCompileAttr << ") is false."; } return !compile; } status = flib_def_->GetAttr(*node, kXlaCompileAttr, &compile); if (status.ok()) { if (!compile) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.h
std::vector<int>* constant_arg_indices, std::vector<int>* resource_arg_indices); // Given a NodeDef `node_def` returns true iff `node_def` has kXlaCompileAttr // set. bool CanCreateXlaKernel(const NodeDef& node_def); // Returns memory types for the input. // `constant_arg_indices` and `resource_arg_indices` are sorted arrays of
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.cc
static bool NodeCanTriggerXlaCompilation(const NodeDef& node) { return node.attr().find(kXlaClusterIdAttr) != node.attr().end() || HasBoolAttr(node, kXlaMustCompileAttr) || HasBoolAttr(node, kXlaCompileAttr) || HasBoolAttr(node, kXlaScopeAttr) || HasBoolAttr(node, kXlaInternalScopeAttr) || ops_triggering_xla_compilation->count(node.op()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 30.3K bytes - Viewed (0)