- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for RunFunctionTf2xlaClusteringBridge (0.45 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v2/cluster_tf_test.cc
TEST_F(FunctionClusterTensorflowDialectTest, ClustersTfReplicatedBridge) { CellReader<int64_t> compilation_status(kCompilationStreamz); TF_ASSERT_OK(CreateMlirModule("empty_func.mlir")); TF_EXPECT_OK(RunFunctionTf2xlaClusteringBridge( *mlir_module_, /*is_supported_by_replicated_brige*/ true, /*is_in_fallback_enabled_mode=*/false)); FuncOp main = mlir_module_->lookupSymbol<mlir::func::FuncOp>("main"); ASSERT_TRUE(main);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/cluster_tf.h
// // Output: Modifies the input module in place with clustered operations. // status - Whether the transformation to cluster the input MLIR module was // successful. tensorflow::Status RunFunctionTf2xlaClusteringBridge( mlir::ModuleOp module, bool is_supported_by_replicated_brige, bool is_in_fallback_enabled_mode, llvm::StringRef module_name = llvm::StringRef()); } // namespace v2 } // namespace tf2xla
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 16 23:11:04 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/cluster_tf.cc
pm, module_name); } void CreateReplicatedClusteringPipelineV2(OpPassManager &pm) { CreateReplicatedClusteringPipeline(pm, /*module_name=*/""); } tensorflow::Status RunFunctionTf2xlaClusteringBridge( ModuleOp module, bool is_supported_by_replicated_brige, bool is_in_fallback_enabled_mode, llvm::StringRef module_name) { std::string device_type = is_supported_by_replicated_brige
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:25:18 UTC 2024 - 8.7K bytes - Viewed (0)