- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for _XlaMustCompile (1.13 sec)
-
tensorflow/compiler/jit/xla_kernel_creator_test.cc
FunctionDef fdef = XTimesY(); (*fdef.mutable_attr())["_XlaMustCompile"] = BoolAttr(true); Init({fdef}); XlaKernelCreator xla_kernel_creator; auto callsite = ToNodeProperties(R"pb( name: 'XTimesY' op: 'XTimesY' input: 'a' input: 'b' )pb"); (*(callsite->node_def.mutable_attr()))["_XlaMustCompile"] = BoolAttr(true); // Note: need to set attribute on the created node.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 01:39:55 UTC 2023 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize_compile_and_replicate_attributes.cc
// attributes into unified attributes. For example, A _tpu_replicate=X // should be replaced with _xla_compile_device_type=TPU and // _replication_info=X attributes by the conversion. An _XlaMustCompile=true // should be replaced with _xla_compile_device_type with the value of device // attribute. #include "llvm/ADT/StringRef.h" #include "llvm/Support/Debug.h" #include "mlir/IR/Builders.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.5K bytes - Viewed (0) -
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";
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
// Name of attribute used to tag operators for compilation with XLA // Implies must-compile semantics: either it will be compiled // with XLA, or an error will be thrown. 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"
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/mlir/tf2xla/internal/mlir_bridge_pass_util.h
#include "tensorflow/core/framework/function.h" namespace tensorflow { // Checks if a graph or reachable functions in the library have any // StatefulPartitionedOps with _XlaMustCompile=true. The function library will // be skipped if nullptr is provided. bool IsSupportedByNonReplicatedBridge( const Graph& graph, const FunctionLibraryDefinition* function_library);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 16:33:22 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize_compile_and_replicate_attributes.mlir
%0 = "tf.StatefulPartitionedCall"(%arg0) {_XlaMustCompile = true, config = "", config_proto = "", device = "/device:CPU:0", executor_type = "", f = @stateful_pcall_func} : (tensor<i32>) -> (tensor<i32>) func.return %0 : tensor<i32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/saved_model/testdata/xla_launch.mlir
%1 = "tf.AddV2"(%arg0, %arg0) {} : (tensor<i32>, tensor<i32>) -> tensor<i32> func.return %1 : tensor<i32> } func.func private @xla_func_0(%arg0: tensor<1x3xf32>, %arg1: tensor<1x3xf32>) -> tensor<1x3xf32> attributes {tf._XlaMustCompile = true, tf._noinline = true, tf._original_func_name = "should_not_be_used"} { %1 = "tf.AddV2"(%arg0, %arg1) : (tensor<1x3xf32>, tensor<1x3xf32>) -> tensor<1x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_cluster_formation.cc
using mlir::func::FuncOp; #define GEN_PASS_DEF_XLACLUSTERFORMATIONPASS #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc" // Outlines partitioned call ops with `_XlaMustCompile` to device clusters. struct XlaClusterFormationPass : public impl::XlaClusterFormationPassBase<XlaClusterFormationPass> { void runOnOperation() override; }; void EncapsulatePartitionedCall(Operation *call_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
for (const Node* node : graph.nodes()) { auto node_op = node->type_string(); if (node_op == kStatefulPartitionedCallOp) { // Functions called by StatefulfulPartitionedCall ops with // _XlaMustCompile=true are compiled by XLA. auto attr = node->attrs().FindByString(std::string(kMustCompileAttr)); if (attr != nullptr && attr->b() == true) { return true; } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0)