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