diff --git a/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/BUILD b/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/BUILD index 7e81deb64a5..0b28d434c7c 100644 --- a/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/BUILD +++ b/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/BUILD @@ -23,7 +23,7 @@ filegroup( data = [ ":importer_test_legacy_reshape", ":importer_test_min_max", - ":schema.fbs", + ":test_schema.fbs", "//tensorflow/compiler/mlir/lite:flatbuffer_to_string", "//tensorflow/compiler/mlir/lite:flatbuffer_translate", "//tensorflow/compiler/mlir/lite:json_to_flatbuffer", diff --git a/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/import_json.json b/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/import_json.json index 0495a141656..f2d275f7ee1 100644 --- a/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/import_json.json +++ b/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/import_json.json @@ -1,4 +1,4 @@ -// RUN: json_to_flatbuffer %p/schema.fbs %s | flatbuffer_translate --tflite-flatbuffer-to-mlir -o - | FileCheck --dump-input-on-failure %s +// RUN: json_to_flatbuffer %p/test_schema.fbs %s | flatbuffer_translate --tflite-flatbuffer-to-mlir -o - | FileCheck --dump-input-on-failure %s // CHECK: %cst = constant unit // CHECK: %[[RES0:.*]] = "tfl.conv_2d"(%arg0, %arg1, %cst) {dilation_h_factor = 1 : i32, dilation_w_factor = 1 : i32, fused_activation_function = "NONE", padding = "SAME", stride_h = 0 : i32, stride_w = 0 : i32} : (tensor<256x32x32x3xf32>, tensor<16x3x3x3xf32>, none) -> tensor<256x32x32x16xf32> diff --git a/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/optional_input.json b/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/optional_input.json index 6f9c262a70f..d6bf73c6c8f 100644 --- a/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/optional_input.json +++ b/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/optional_input.json @@ -1,4 +1,4 @@ -// RUN: json_to_flatbuffer %p/schema.fbs %s | flatbuffer_translate --tflite-flatbuffer-to-mlir -o - | FileCheck --dump-input-on-failure %s +// RUN: json_to_flatbuffer %p/test_schema.fbs %s | flatbuffer_translate --tflite-flatbuffer-to-mlir -o - | FileCheck --dump-input-on-failure %s // This test is to test that if the flatbuffer omits the last optional input `bias` of tfl.conv_2d op, the flatbuffer_importer will automatically adds `none` value to tfl.conv_2d. diff --git a/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/schema.fbs b/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/test_schema.fbs similarity index 100% rename from tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/schema.fbs rename to tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/test_schema.fbs