Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for BuildOpGraphWithOutputShapes (0.35 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util_test.cc 

      TF_ASSERT_OK_AND_ASSIGN(auto graph, BuildOpGraphWithOutputShapes());

  auto build_result = BuildHloFromGraph(*graph, /*use_output_shapes=*/false);
  TF_ASSERT_OK(build_result);
}

TEST(CompileMlirUtil, UsesIncorrectOutputShapesWhenPresent) {
  TF_ASSERT_OK_AND_ASSIGN(auto graph, BuildOpGraphWithOutputShapes());

  auto build_result = BuildHloFromGraph(*graph, /*use_output_shapes=*/true);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 25 19:54:38 UTC 2024
    - 9.7K bytes
    - Viewed (0)
Back to top