Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for RunFunctionTf2xlaClusteringBridge (0.34 sec)

  1. 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)
Back to top