- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for _XlaMustCompile (0.21 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/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) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc
if (isa<TPUPartitionedCallOp>(call)) return false; if (isa<RemoteCallOp>(call)) return false; // Maintain inlining for `tf.function`s with jit_compile option. if (callable->hasAttr("tf._XlaMustCompile")) return true; auto noinline_attr_name = absl::StrCat("tf.", tensorflow::kNoInlineAttr); if (auto noinline_attr = callable->getAttrOfType<BoolAttr>(noinline_attr_name))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir
return %x : tensor<i32> } // ----- // Test for XlaLaunch 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) {__op_key = 0: i32} : (tensor<1x3xf32>, tensor<1x3xf32>) -> tensor<1x3xf32> func.return %1 : tensor<1x3xf32> }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
// tf.StatefulPartitionedCall ops with and without // _tpu_replicate attributes may exist in the same graph. Ops without // the attribute but with _XlaMustCompile=true would have // _xla_compile_device_type="" after // CanonicalizeCompileAndReplicateAttributesPass. Skip empty value here. if (!device_type_attr.getValue().empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0)