- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for IsSupportedByNonReplicatedBridge (0.36 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.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); // Checks if a graph or reachable functions in the library have any ops with
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 NodeDef& node_def : func_def->node_def()) { if (node_def.op() == kPartitionedCall) return true; } } return false; } } // namespace bool IsSupportedByNonReplicatedBridge( const Graph& graph, const FunctionLibraryDefinition* function_library) { return IsNonReplicatedGraph(graph, function_library) && HasPsWithResourceVariable(graph); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0)