Have you already tried also passing it like in tfcompile_flags:

  

      github.com
  

  
    tensorflow/tensorflow/blob/master/tensorflow/compiler/aot/tests/BUILD#L238 1



      
      
          tf_library(
              name = "test_graph_tfmatmulandadd",
              testonly = 1,
              config = "test_graph_tfmatmulandadd.config.pbtxt",
              cpp_class = "::foo::bar::MatMulAndAddComp",
              graph = "test_graph_tfmatmulandadd.pb",
              mlir_components = "None",
              tags = [
                  "manual",
              ],
              tfcompile_flags = "--gen_name_to_index --gen_program_shape",
          )
          
          
tf_library(
              name = "test_graph_tfmatmulandadd_with_profiling",
              testonly = 1,
              config = "test_graph_tfmatmulandadd.config.pbtxt",
              cpp_class = "MatMulAndAddCompWithProfiling",
              enable_xla_hlo_profiling = True,
              graph = "test_graph_tfmatmulandadd.pb",
              mlir_components = "None",