TF-OPT issue: StatelessRandomUniformV2 requires Compile Time Constant  
@monowaranjum,
I tried to execute the mentioned above on google colab with the latest stable tensorflow version and it was executed without any issue/error. Kindly find the gist of it [here](https://colab.research.google.com/gist/tilakrayal/e5a090e8bcc172e2145e200e6ae1534c/untitled1527.ipynb). Thank you! This issue is stale because it has been open for 7 days with no activity. It will be closed if no further activity occurs. Thank you. Are you satisfied with the resolution of your issue?
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=Yes&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/62442">Yes</a>
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=No&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/62442">No</a>

### Issue type

Bug

### Have you reproduced the bug with TensorFlow Nightly?

Yes

### Source

source

### TensorFlow version

2.16.0-dev20231013

### Custom code

Yes

### OS platform and distribution

Ubuntu 22.04

### Mobile device

_No response_

### Python version

3.10

### Bazel version

5.1.1

### GCC/compiler version

g++-11

### CUDA/cuDNN version

_No response_

### GPU model and memory

_No response_

### Current behavior?

I built a custom model which classifies the mnist digits. Then used ```tf-mlir-translate``` to convert the saved model to tf dialect mlir. Then I tried to use the ```tf-opt``` with ```--tf-to-hlo-pipeline``` pass to get the hlo dialect from tensorflow dialect. It throws an error that says: 

```
W tensorflow/core/framework/op_kernel.cc:1839] OP_REQUIRES failed at stateless_random_ops_v2.cc:93 : INVALID_ARGUMENT: Input 0 to node `tf.StatelessRandomUniformV2` with op StatelessRandomUniformV2 must be a compile-time constant.

XLA compilation requires that operator arguments that represent shapes or dimensions be evaluated to concrete values at compile time. This error means that a shape or dimension argument could not be evaluated at compile time, usually because the value of the argument depends on a parameter to the computation, on a variable, or on a stateful operation such as a random number generator.

/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:21:59: remark: compilation to HLO failed: INVALID_ARGUMENT: Input 0 to node `tf.StatelessRandomUniformV2` with op StatelessRandomUniformV2 must be a compile-time constant.

XLA compilation requires that operator arguments that represent shapes or dimensions be evaluated to concrete values at compile time. This error means that a shape or dimension argument could not be evaluated at compile time, usually because the value of the argument depends on a parameter to the computation, on a variable, or on a stateful operation such as a random number generator.

```

### Standalone code to reproduce the issue

```shell
# Here is the model code: 

import os
from tensorflow import mlir
import tensorflow_hub as hub
import tensorflow as tf
import keras
os.environ["TFHUB_DOWNLOAD_PROGRESS"] = "True"

def get_model():

    g = tf.Graph()
    with g.as_default():
        tf.random.set_seed(42)
        model = keras.models.Sequential()
        model.add(keras.layers.Input((32,784)))
        model.add(keras.layers.Dense(64, activation='relu'))
        model.add(keras.layers.Dense(10, activation='softmax'))
        
        tf.io.write_graph(g.as_graph_def(), "logdir" , "mnist.pb")
    
    return model



get_model()

# Here is the command I ran for tf-mlir-translate

./tf-mlir-translate --graphdef-to-mlir --tf-enable-shape-inference-on-import \ /home/rashik/Documents/tensorfow_mlir_test/mnist.pb -o \ /home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir

# Here is the command I ran for tf-opt 

./tf-opt --tf-to-hlo-pipeline /home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir -o /home/rashik/Documents/tensorfow_mlir_test/hlo_dialect.mlir
```


### Relevant log output

```shell
// Output from tf-mlir-translate: 
module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 1647 : i32}} {
  func.func @main() {
    tf_executor.graph {
      %outputs, %control = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense/bias"}> {_class = ["loc:@dense/bias"], allowed_devices = [], debug_name = "dense/bias/", device = ""} : () -> tensor<!tf_type.resource<tensor<64xf32>>>
      %outputs_0, %control_1 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<64xf32>
      %outputs_2, %control_3 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<i1>
      %outputs_4, %control_5 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<64xf32>
      %outputs_6, %control_7 = tf_executor.island wraps "tf.Const"() <{value = dense<0.000000e+00> : tensor<64xf32>}> {_class = ["loc:@dense/bias"], device = ""} : () -> tensor<64xf32>
      %control_8 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs, %outputs_6) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>, tensor<64xf32>) -> ()
      %outputs_9, %control_10 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense/kernel"}> {_class = ["loc:@dense/kernel"], allowed_devices = [], debug_name = "dense/kernel/", device = ""} : () -> tensor<!tf_type.resource<tensor<784x64xf32>>>
      %outputs_11, %control_12 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<784x64xf32>
      %outputs_13, %control_14 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<i1>
      %outputs_15, %control_16 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<784x64xf32>
      %outputs_17, %control_18 = tf_executor.island wraps "tf.Const"() <{value = dense<[835938099, 0]> : tensor<2xi32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<2xi32>
      %outputs_19:2, %control_20 = tf_executor.island wraps "tf.StatelessRandomGetKeyCounter"(%outputs_17) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<2xi32>) -> (tensor<1xui64>, tensor<2xui64>)
      %outputs_21, %control_22 = tf_executor.island wraps "tf.Const"() <{value = dense<3> : tensor<i32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<i32>
      %outputs_23, %control_24 = tf_executor.island wraps "tf.Const"() <{value = dense<0.0841158255> : tensor<f32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<f32>
      %outputs_25, %control_26 = tf_executor.island wraps "tf.Const"() <{value = dense<-0.0841158255> : tensor<f32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<f32>
      %outputs_27, %control_28 = tf_executor.island wraps "tf.Sub"(%outputs_23, %outputs_25) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<f32>, tensor<f32>) -> tensor<f32>
      %outputs_29, %control_30 = tf_executor.island wraps "tf.Const"() <{value = dense<[784, 64]> : tensor<2xi32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<2xi32>
      %outputs_31, %control_32 = tf_executor.island wraps "tf.StatelessRandomUniformV2"(%outputs_29, %outputs_19#0, %outputs_19#1, %outputs_21) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<784x64xf32>
      %outputs_33, %control_34 = tf_executor.island wraps "tf.Mul"(%outputs_31, %outputs_27) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<784x64xf32>, tensor<f32>) -> tensor<784x64xf32>
      %outputs_35, %control_36 = tf_executor.island wraps "tf.AddV2"(%outputs_33, %outputs_25) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<784x64xf32>, tensor<f32>) -> tensor<784x64xf32>
      %control_37 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_9, %outputs_35) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>, tensor<784x64xf32>) -> ()
      %outputs_38, %control_39 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense_1/bias"}> {_class = ["loc:@dense_1/bias"], allowed_devices = [], debug_name = "dense_1/bias/", device = ""} : () -> tensor<!tf_type.resource<tensor<10xf32>>>
      %outputs_40, %control_41 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<10xf32>
      %outputs_42, %control_43 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<i1>
      %outputs_44, %control_45 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<10xf32>
      %outputs_46, %control_47 = tf_executor.island wraps "tf.Const"() <{value = dense<0.000000e+00> : tensor<10xf32>}> {_class = ["loc:@dense_1/bias"], device = ""} : () -> tensor<10xf32>
      %control_48 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_38, %outputs_46) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>, tensor<10xf32>) -> ()
      %outputs_49, %control_50 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense_1/kernel"}> {_class = ["loc:@dense_1/kernel"], allowed_devices = [], debug_name = "dense_1/kernel/", device = ""} : () -> tensor<!tf_type.resource<tensor<64x10xf32>>>
      %outputs_51, %control_52 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<64x10xf32>
      %outputs_53, %control_54 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<i1>
      %outputs_55, %control_56 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<64x10xf32>
      %outputs_57, %control_58 = tf_executor.island wraps "tf.Const"() <{value = dense<[9356290, 0]> : tensor<2xi32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<2xi32>
      %outputs_59:2, %control_60 = tf_executor.island wraps "tf.StatelessRandomGetKeyCounter"(%outputs_57) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<2xi32>) -> (tensor<1xui64>, tensor<2xui64>)
      %outputs_61, %control_62 = tf_executor.island wraps "tf.Const"() <{value = dense<3> : tensor<i32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<i32>
      %outputs_63, %control_64 = tf_executor.island wraps "tf.Const"() <{value = dense<0.284747392> : tensor<f32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<f32>
      %outputs_65, %control_66 = tf_executor.island wraps "tf.Const"() <{value = dense<-0.284747392> : tensor<f32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<f32>
      %outputs_67, %control_68 = tf_executor.island wraps "tf.Sub"(%outputs_63, %outputs_65) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<f32>, tensor<f32>) -> tensor<f32>
      %outputs_69, %control_70 = tf_executor.island wraps "tf.Const"() <{value = dense<[64, 10]> : tensor<2xi32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<2xi32>
      %outputs_71, %control_72 = tf_executor.island wraps "tf.StatelessRandomUniformV2"(%outputs_69, %outputs_59#0, %outputs_59#1, %outputs_61) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<64x10xf32>
      %outputs_73, %control_74 = tf_executor.island wraps "tf.Mul"(%outputs_71, %outputs_67) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<64x10xf32>, tensor<f32>) -> tensor<64x10xf32>
      %outputs_75, %control_76 = tf_executor.island wraps "tf.AddV2"(%outputs_73, %outputs_65) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<64x10xf32>, tensor<f32>) -> tensor<64x10xf32>
      %control_77 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_49, %outputs_75) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>, tensor<64x10xf32>) -> ()
      %outputs_78, %control_79 = tf_executor.island wraps "tf.Placeholder"() {device = "", shape = #tf_type.shape<?x784>} : () -> tensor<?x784xf32>
      %outputs_80, %control_81 = tf_executor.island wraps "tf.MatMul"(%outputs_78, %outputs_11) <{transpose_a = false, transpose_b = false}> {device = ""} : (tensor<?x784xf32>, tensor<784x64xf32>) -> tensor<?x64xf32>
      %outputs_82, %control_83 = tf_executor.island wraps "tf.BiasAdd"(%outputs_80, %outputs_0) <{data_format = "NHWC"}> {device = ""} : (tensor<?x64xf32>, tensor<64xf32>) -> tensor<?x64xf32>
      %outputs_84, %control_85 = tf_executor.island wraps "tf.Relu"(%outputs_82) {device = ""} : (tensor<?x64xf32>) -> tensor<?x64xf32>
      %outputs_86, %control_87 = tf_executor.island wraps "tf.MatMul"(%outputs_84, %outputs_51) <{transpose_a = false, transpose_b = false}> {device = ""} : (tensor<?x64xf32>, tensor<64x10xf32>) -> tensor<?x10xf32>
      %outputs_88, %control_89 = tf_executor.island wraps "tf.BiasAdd"(%outputs_86, %outputs_40) <{data_format = "NHWC"}> {device = ""} : (tensor<?x10xf32>, tensor<10xf32>) -> tensor<?x10xf32>
      %outputs_90, %control_91 = tf_executor.island wraps "tf.Softmax"(%outputs_88) {device = ""} : (tensor<?x10xf32>) -> tensor<?x10xf32>
      tf_executor.fetch
    }
    return
  }
}




************************* Log Output ************************

2023-11-20 11:15:08.462092: W tensorflow/core/framework/op_kernel.cc:1839] OP_REQUIRES failed at stateless_random_ops_v2.cc:93 : INVALID_ARGUMENT: Input 0 to node `tf.StatelessRandomUniformV2` with op StatelessRandomUniformV2 must be a compile-time constant.

XLA compilation requires that operator arguments that represent shapes or dimensions be evaluated to concrete values at compile time. This error means that a shape or dimension argument could not be evaluated at compile time, usually because the value of the argument depends on a parameter to the computation, on a variable, or on a stateful operation such as a random number generator.

/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:21:59: remark: compilation to HLO failed: INVALID_ARGUMENT: Input 0 to node `tf.StatelessRandomUniformV2` with op StatelessRandomUniformV2 must be a compile-time constant.

XLA compilation requires that operator arguments that represent shapes or dimensions be evaluated to concrete values at compile time. This error means that a shape or dimension argument could not be evaluated at compile time, usually because the value of the argument depends on a parameter to the computation, on a variable, or on a stateful operation such as a random number generator.

      %outputs_31, %control_32 = tf_executor.island wraps "tf.StatelessRandomUniformV2"(%outputs_29, %outputs_19#0, %outputs_19#1, %outputs_21) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:21:59: note: see current operation: %0 = "tf.StatelessRandomUniformV2"(%outputs_29, %outputs_19#0, %outputs_19#1, %outputs_21) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<784x64xf32>
2023-11-20 11:15:08.463988: W tensorflow/core/framework/op_kernel.cc:1839] OP_REQUIRES failed at stateless_random_ops_v2.cc:93 : INVALID_ARGUMENT: Input 0 to node `tf.StatelessRandomUniformV2` with op StatelessRandomUniformV2 must be a compile-time constant.

XLA compilation requires that operator arguments that represent shapes or dimensions be evaluated to concrete values at compile time. This error means that a shape or dimension argument could not be evaluated at compile time, usually because the value of the argument depends on a parameter to the computation, on a variable, or on a stateful operation such as a random number generator.

/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:42:59: remark: compilation to HLO failed: INVALID_ARGUMENT: Input 0 to node `tf.StatelessRandomUniformV2` with op StatelessRandomUniformV2 must be a compile-time constant.

XLA compilation requires that operator arguments that represent shapes or dimensions be evaluated to concrete values at compile time. This error means that a shape or dimension argument could not be evaluated at compile time, usually because the value of the argument depends on a parameter to the computation, on a variable, or on a stateful operation such as a random number generator.

      %outputs_71, %control_72 = tf_executor.island wraps "tf.StatelessRandomUniformV2"(%outputs_69, %outputs_59#0, %outputs_59#1, %outputs_61) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:42:59: note: see current operation: %0 = "tf.StatelessRandomUniformV2"(%outputs_69, %outputs_59#0, %outputs_59#1, %outputs_61) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<64x10xf32>
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:4:53: error: Could not legalize op: tf.VarHandleOp
      %outputs, %control = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense/bias"}> {_class = ["loc:@dense/bias"], allowed_devices = [], debug_name = "dense/bias/", device = ""} : () -> tensor<!tf_type.resource<tensor<64xf32>>>
                                                    ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:4:53: error: Could not legalize op: tf_executor.yield
      %outputs, %control = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense/bias"}> {_class = ["loc:@dense/bias"], allowed_devices = [], debug_name = "dense/bias/", device = ""} : () -> tensor<!tf_type.resource<tensor<64xf32>>>
                                                    ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:4:53: error: Could not legalize op: tf_executor.island
      %outputs, %control = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense/bias"}> {_class = ["loc:@dense/bias"], allowed_devices = [], debug_name = "dense/bias/", device = ""} : () -> tensor<!tf_type.resource<tensor<64xf32>>>
                                                    ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:5:57: error: Could not legalize op: tf.ReadVariableOp
      %outputs_0, %control_1 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<64xf32>
                                                        ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:5:57: error: Could not legalize op: tf_executor.yield
      %outputs_0, %control_1 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<64xf32>
                                                        ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:5:57: error: Could not legalize op: tf_executor.island
      %outputs_0, %control_1 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<64xf32>
                                                        ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:6:57: error: Could not legalize op: tf.VarIsInitializedOp
      %outputs_2, %control_3 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<i1>
                                                        ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:6:57: error: Could not legalize op: tf_executor.yield
      %outputs_2, %control_3 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<i1>
                                                        ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:6:57: error: Could not legalize op: tf_executor.island
      %outputs_2, %control_3 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<i1>
                                                        ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:7:57: error: Could not legalize op: tf.ReadVariableOp
      %outputs_4, %control_5 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<64xf32>
                                                        ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:7:57: error: Could not legalize op: tf_executor.yield
      %outputs_4, %control_5 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<64xf32>
                                                        ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:7:57: error: Could not legalize op: tf_executor.island
      %outputs_4, %control_5 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs) {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>) -> tensor<64xf32>
                                                        ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:8:57: error: Could not legalize op: tf_executor.yield
      %outputs_6, %control_7 = tf_executor.island wraps "tf.Const"() <{value = dense<0.000000e+00> : tensor<64xf32>}> {_class = ["loc:@dense/bias"], device = ""} : () -> tensor<64xf32>
                                                        ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:8:57: error: Could not legalize op: tf_executor.island
      %outputs_6, %control_7 = tf_executor.island wraps "tf.Const"() <{value = dense<0.000000e+00> : tensor<64xf32>}> {_class = ["loc:@dense/bias"], device = ""} : () -> tensor<64xf32>
                                                        ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:9:45: error: Could not legalize op: tf.AssignVariableOp
      %control_8 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs, %outputs_6) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>, tensor<64xf32>) -> ()
                                            ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:9:45: error: Could not legalize op: tf_executor.yield
      %control_8 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs, %outputs_6) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>, tensor<64xf32>) -> ()
                                            ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:9:45: error: Could not legalize op: tf_executor.island
      %control_8 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs, %outputs_6) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<64xf32>>>, tensor<64xf32>) -> ()
                                            ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:10:58: error: Could not legalize op: tf.VarHandleOp
      %outputs_9, %control_10 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense/kernel"}> {_class = ["loc:@dense/kernel"], allowed_devices = [], debug_name = "dense/kernel/", device = ""} : () -> tensor<!tf_type.resource<tensor<784x64xf32>>>
                                                         ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:10:58: error: Could not legalize op: tf_executor.yield
      %outputs_9, %control_10 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense/kernel"}> {_class = ["loc:@dense/kernel"], allowed_devices = [], debug_name = "dense/kernel/", device = ""} : () -> tensor<!tf_type.resource<tensor<784x64xf32>>>
                                                         ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:10:58: error: Could not legalize op: tf_executor.island
      %outputs_9, %control_10 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense/kernel"}> {_class = ["loc:@dense/kernel"], allowed_devices = [], debug_name = "dense/kernel/", device = ""} : () -> tensor<!tf_type.resource<tensor<784x64xf32>>>
                                                         ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:11:59: error: Could not legalize op: tf.ReadVariableOp
      %outputs_11, %control_12 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:11:59: error: Could not legalize op: tf_executor.yield
      %outputs_11, %control_12 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:11:59: error: Could not legalize op: tf_executor.island
      %outputs_11, %control_12 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:12:59: error: Could not legalize op: tf.VarIsInitializedOp
      %outputs_13, %control_14 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<i1>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:12:59: error: Could not legalize op: tf_executor.yield
      %outputs_13, %control_14 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<i1>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:12:59: error: Could not legalize op: tf_executor.island
      %outputs_13, %control_14 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<i1>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:13:59: error: Could not legalize op: tf.ReadVariableOp
      %outputs_15, %control_16 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:13:59: error: Could not legalize op: tf_executor.yield
      %outputs_15, %control_16 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:13:59: error: Could not legalize op: tf_executor.island
      %outputs_15, %control_16 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_9) {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:14:59: error: Could not legalize op: tf_executor.yield
      %outputs_17, %control_18 = tf_executor.island wraps "tf.Const"() <{value = dense<[835938099, 0]> : tensor<2xi32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<2xi32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:14:59: error: Could not legalize op: tf_executor.island
      %outputs_17, %control_18 = tf_executor.island wraps "tf.Const"() <{value = dense<[835938099, 0]> : tensor<2xi32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<2xi32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:15:61: error: Could not legalize op: tf_executor.yield
      %outputs_19:2, %control_20 = tf_executor.island wraps "tf.StatelessRandomGetKeyCounter"(%outputs_17) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<2xi32>) -> (tensor<1xui64>, tensor<2xui64>)
                                                            ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:15:61: error: Could not legalize op: tf_executor.island
      %outputs_19:2, %control_20 = tf_executor.island wraps "tf.StatelessRandomGetKeyCounter"(%outputs_17) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<2xi32>) -> (tensor<1xui64>, tensor<2xui64>)
                                                            ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:16:59: error: Could not legalize op: tf_executor.yield
      %outputs_21, %control_22 = tf_executor.island wraps "tf.Const"() <{value = dense<3> : tensor<i32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<i32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:16:59: error: Could not legalize op: tf_executor.island
      %outputs_21, %control_22 = tf_executor.island wraps "tf.Const"() <{value = dense<3> : tensor<i32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<i32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:17:59: error: Could not legalize op: tf_executor.yield
      %outputs_23, %control_24 = tf_executor.island wraps "tf.Const"() <{value = dense<0.0841158255> : tensor<f32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:17:59: error: Could not legalize op: tf_executor.island
      %outputs_23, %control_24 = tf_executor.island wraps "tf.Const"() <{value = dense<0.0841158255> : tensor<f32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:18:59: error: Could not legalize op: tf_executor.yield
      %outputs_25, %control_26 = tf_executor.island wraps "tf.Const"() <{value = dense<-0.0841158255> : tensor<f32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:18:59: error: Could not legalize op: tf_executor.island
      %outputs_25, %control_26 = tf_executor.island wraps "tf.Const"() <{value = dense<-0.0841158255> : tensor<f32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:19:59: error: Could not legalize op: tf_executor.yield
      %outputs_27, %control_28 = tf_executor.island wraps "tf.Sub"(%outputs_23, %outputs_25) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<f32>, tensor<f32>) -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:19:59: error: Could not legalize op: tf_executor.island
      %outputs_27, %control_28 = tf_executor.island wraps "tf.Sub"(%outputs_23, %outputs_25) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<f32>, tensor<f32>) -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:20:59: error: Could not legalize op: tf_executor.yield
      %outputs_29, %control_30 = tf_executor.island wraps "tf.Const"() <{value = dense<[784, 64]> : tensor<2xi32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<2xi32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:20:59: error: Could not legalize op: tf_executor.island
      %outputs_29, %control_30 = tf_executor.island wraps "tf.Const"() <{value = dense<[784, 64]> : tensor<2xi32>}> {_class = ["loc:@dense/kernel"], device = ""} : () -> tensor<2xi32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:21:59: error: Could not legalize op: tf.StatelessRandomUniformV2
      %outputs_31, %control_32 = tf_executor.island wraps "tf.StatelessRandomUniformV2"(%outputs_29, %outputs_19#0, %outputs_19#1, %outputs_21) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:21:59: error: Could not legalize op: tf_executor.yield
      %outputs_31, %control_32 = tf_executor.island wraps "tf.StatelessRandomUniformV2"(%outputs_29, %outputs_19#0, %outputs_19#1, %outputs_21) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:21:59: error: Could not legalize op: tf_executor.island
      %outputs_31, %control_32 = tf_executor.island wraps "tf.StatelessRandomUniformV2"(%outputs_29, %outputs_19#0, %outputs_19#1, %outputs_21) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:22:59: error: Could not legalize op: tf_executor.yield
      %outputs_33, %control_34 = tf_executor.island wraps "tf.Mul"(%outputs_31, %outputs_27) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<784x64xf32>, tensor<f32>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:22:59: error: Could not legalize op: tf_executor.island
      %outputs_33, %control_34 = tf_executor.island wraps "tf.Mul"(%outputs_31, %outputs_27) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<784x64xf32>, tensor<f32>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:23:59: error: Could not legalize op: tf_executor.yield
      %outputs_35, %control_36 = tf_executor.island wraps "tf.AddV2"(%outputs_33, %outputs_25) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<784x64xf32>, tensor<f32>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:23:59: error: Could not legalize op: tf_executor.island
      %outputs_35, %control_36 = tf_executor.island wraps "tf.AddV2"(%outputs_33, %outputs_25) {_class = ["loc:@dense/kernel"], device = ""} : (tensor<784x64xf32>, tensor<f32>) -> tensor<784x64xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:24:46: error: Could not legalize op: tf.AssignVariableOp
      %control_37 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_9, %outputs_35) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>, tensor<784x64xf32>) -> ()
                                             ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:24:46: error: Could not legalize op: tf_executor.yield
      %control_37 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_9, %outputs_35) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>, tensor<784x64xf32>) -> ()
                                             ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:24:46: error: Could not legalize op: tf_executor.island
      %control_37 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_9, %outputs_35) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<784x64xf32>>>, tensor<784x64xf32>) -> ()
                                             ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:25:59: error: Could not legalize op: tf.VarHandleOp
      %outputs_38, %control_39 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense_1/bias"}> {_class = ["loc:@dense_1/bias"], allowed_devices = [], debug_name = "dense_1/bias/", device = ""} : () -> tensor<!tf_type.resource<tensor<10xf32>>>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:25:59: error: Could not legalize op: tf_executor.yield
      %outputs_38, %control_39 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense_1/bias"}> {_class = ["loc:@dense_1/bias"], allowed_devices = [], debug_name = "dense_1/bias/", device = ""} : () -> tensor<!tf_type.resource<tensor<10xf32>>>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:25:59: error: Could not legalize op: tf_executor.island
      %outputs_38, %control_39 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense_1/bias"}> {_class = ["loc:@dense_1/bias"], allowed_devices = [], debug_name = "dense_1/bias/", device = ""} : () -> tensor<!tf_type.resource<tensor<10xf32>>>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:26:59: error: Could not legalize op: tf.ReadVariableOp
      %outputs_40, %control_41 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:26:59: error: Could not legalize op: tf_executor.yield
      %outputs_40, %control_41 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:26:59: error: Could not legalize op: tf_executor.island
      %outputs_40, %control_41 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:27:59: error: Could not legalize op: tf.VarIsInitializedOp
      %outputs_42, %control_43 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<i1>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:27:59: error: Could not legalize op: tf_executor.yield
      %outputs_42, %control_43 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<i1>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:27:59: error: Could not legalize op: tf_executor.island
      %outputs_42, %control_43 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<i1>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:28:59: error: Could not legalize op: tf.ReadVariableOp
      %outputs_44, %control_45 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:28:59: error: Could not legalize op: tf_executor.yield
      %outputs_44, %control_45 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:28:59: error: Could not legalize op: tf_executor.island
      %outputs_44, %control_45 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_38) {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>) -> tensor<10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:29:59: error: Could not legalize op: tf_executor.yield
      %outputs_46, %control_47 = tf_executor.island wraps "tf.Const"() <{value = dense<0.000000e+00> : tensor<10xf32>}> {_class = ["loc:@dense_1/bias"], device = ""} : () -> tensor<10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:29:59: error: Could not legalize op: tf_executor.island
      %outputs_46, %control_47 = tf_executor.island wraps "tf.Const"() <{value = dense<0.000000e+00> : tensor<10xf32>}> {_class = ["loc:@dense_1/bias"], device = ""} : () -> tensor<10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:30:46: error: Could not legalize op: tf.AssignVariableOp
      %control_48 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_38, %outputs_46) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>, tensor<10xf32>) -> ()
                                             ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:30:46: error: Could not legalize op: tf_executor.yield
      %control_48 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_38, %outputs_46) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>, tensor<10xf32>) -> ()
                                             ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:30:46: error: Could not legalize op: tf_executor.island
      %control_48 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_38, %outputs_46) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<10xf32>>>, tensor<10xf32>) -> ()
                                             ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:31:59: error: Could not legalize op: tf.VarHandleOp
      %outputs_49, %control_50 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense_1/kernel"}> {_class = ["loc:@dense_1/kernel"], allowed_devices = [], debug_name = "dense_1/kernel/", device = ""} : () -> tensor<!tf_type.resource<tensor<64x10xf32>>>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:31:59: error: Could not legalize op: tf_executor.yield
      %outputs_49, %control_50 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense_1/kernel"}> {_class = ["loc:@dense_1/kernel"], allowed_devices = [], debug_name = "dense_1/kernel/", device = ""} : () -> tensor<!tf_type.resource<tensor<64x10xf32>>>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:31:59: error: Could not legalize op: tf_executor.island
      %outputs_49, %control_50 = tf_executor.island wraps "tf.VarHandleOp"() <{container = "", shared_name = "dense_1/kernel"}> {_class = ["loc:@dense_1/kernel"], allowed_devices = [], debug_name = "dense_1/kernel/", device = ""} : () -> tensor<!tf_type.resource<tensor<64x10xf32>>>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:32:59: error: Could not legalize op: tf.ReadVariableOp
      %outputs_51, %control_52 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:32:59: error: Could not legalize op: tf_executor.yield
      %outputs_51, %control_52 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:32:59: error: Could not legalize op: tf_executor.island
      %outputs_51, %control_52 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:33:59: error: Could not legalize op: tf.VarIsInitializedOp
      %outputs_53, %control_54 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<i1>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:33:59: error: Could not legalize op: tf_executor.yield
      %outputs_53, %control_54 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<i1>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:33:59: error: Could not legalize op: tf_executor.island
      %outputs_53, %control_54 = tf_executor.island wraps "tf.VarIsInitializedOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<i1>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:34:59: error: Could not legalize op: tf.ReadVariableOp
      %outputs_55, %control_56 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:34:59: error: Could not legalize op: tf_executor.yield
      %outputs_55, %control_56 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:34:59: error: Could not legalize op: tf_executor.island
      %outputs_55, %control_56 = tf_executor.island wraps "tf.ReadVariableOp"(%outputs_49) {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:35:59: error: Could not legalize op: tf_executor.yield
      %outputs_57, %control_58 = tf_executor.island wraps "tf.Const"() <{value = dense<[9356290, 0]> : tensor<2xi32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<2xi32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:35:59: error: Could not legalize op: tf_executor.island
      %outputs_57, %control_58 = tf_executor.island wraps "tf.Const"() <{value = dense<[9356290, 0]> : tensor<2xi32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<2xi32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:36:61: error: Could not legalize op: tf_executor.yield
      %outputs_59:2, %control_60 = tf_executor.island wraps "tf.StatelessRandomGetKeyCounter"(%outputs_57) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<2xi32>) -> (tensor<1xui64>, tensor<2xui64>)
                                                            ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:36:61: error: Could not legalize op: tf_executor.island
      %outputs_59:2, %control_60 = tf_executor.island wraps "tf.StatelessRandomGetKeyCounter"(%outputs_57) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<2xi32>) -> (tensor<1xui64>, tensor<2xui64>)
                                                            ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:37:59: error: Could not legalize op: tf_executor.yield
      %outputs_61, %control_62 = tf_executor.island wraps "tf.Const"() <{value = dense<3> : tensor<i32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<i32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:37:59: error: Could not legalize op: tf_executor.island
      %outputs_61, %control_62 = tf_executor.island wraps "tf.Const"() <{value = dense<3> : tensor<i32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<i32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:38:59: error: Could not legalize op: tf_executor.yield
      %outputs_63, %control_64 = tf_executor.island wraps "tf.Const"() <{value = dense<0.284747392> : tensor<f32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:38:59: error: Could not legalize op: tf_executor.island
      %outputs_63, %control_64 = tf_executor.island wraps "tf.Const"() <{value = dense<0.284747392> : tensor<f32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:39:59: error: Could not legalize op: tf_executor.yield
      %outputs_65, %control_66 = tf_executor.island wraps "tf.Const"() <{value = dense<-0.284747392> : tensor<f32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:39:59: error: Could not legalize op: tf_executor.island
      %outputs_65, %control_66 = tf_executor.island wraps "tf.Const"() <{value = dense<-0.284747392> : tensor<f32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:40:59: error: Could not legalize op: tf_executor.yield
      %outputs_67, %control_68 = tf_executor.island wraps "tf.Sub"(%outputs_63, %outputs_65) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<f32>, tensor<f32>) -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:40:59: error: Could not legalize op: tf_executor.island
      %outputs_67, %control_68 = tf_executor.island wraps "tf.Sub"(%outputs_63, %outputs_65) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<f32>, tensor<f32>) -> tensor<f32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:41:59: error: Could not legalize op: tf_executor.yield
      %outputs_69, %control_70 = tf_executor.island wraps "tf.Const"() <{value = dense<[64, 10]> : tensor<2xi32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<2xi32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:41:59: error: Could not legalize op: tf_executor.island
      %outputs_69, %control_70 = tf_executor.island wraps "tf.Const"() <{value = dense<[64, 10]> : tensor<2xi32>}> {_class = ["loc:@dense_1/kernel"], device = ""} : () -> tensor<2xi32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:42:59: error: Could not legalize op: tf.StatelessRandomUniformV2
      %outputs_71, %control_72 = tf_executor.island wraps "tf.StatelessRandomUniformV2"(%outputs_69, %outputs_59#0, %outputs_59#1, %outputs_61) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:42:59: error: Could not legalize op: tf_executor.yield
      %outputs_71, %control_72 = tf_executor.island wraps "tf.StatelessRandomUniformV2"(%outputs_69, %outputs_59#0, %outputs_59#1, %outputs_61) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:42:59: error: Could not legalize op: tf_executor.island
      %outputs_71, %control_72 = tf_executor.island wraps "tf.StatelessRandomUniformV2"(%outputs_69, %outputs_59#0, %outputs_59#1, %outputs_61) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<2xi32>, tensor<1xui64>, tensor<2xui64>, tensor<i32>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:43:59: error: Could not legalize op: tf_executor.yield
      %outputs_73, %control_74 = tf_executor.island wraps "tf.Mul"(%outputs_71, %outputs_67) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<64x10xf32>, tensor<f32>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:43:59: error: Could not legalize op: tf_executor.island
      %outputs_73, %control_74 = tf_executor.island wraps "tf.Mul"(%outputs_71, %outputs_67) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<64x10xf32>, tensor<f32>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:44:59: error: Could not legalize op: tf_executor.yield
      %outputs_75, %control_76 = tf_executor.island wraps "tf.AddV2"(%outputs_73, %outputs_65) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<64x10xf32>, tensor<f32>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:44:59: error: Could not legalize op: tf_executor.island
      %outputs_75, %control_76 = tf_executor.island wraps "tf.AddV2"(%outputs_73, %outputs_65) {_class = ["loc:@dense_1/kernel"], device = ""} : (tensor<64x10xf32>, tensor<f32>) -> tensor<64x10xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:45:46: error: Could not legalize op: tf.AssignVariableOp
      %control_77 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_49, %outputs_75) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>, tensor<64x10xf32>) -> ()
                                             ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:45:46: error: Could not legalize op: tf_executor.yield
      %control_77 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_49, %outputs_75) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>, tensor<64x10xf32>) -> ()
                                             ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:45:46: error: Could not legalize op: tf_executor.island
      %control_77 = tf_executor.island wraps "tf.AssignVariableOp"(%outputs_49, %outputs_75) <{validate_shape = false}> {device = ""} : (tensor<!tf_type.resource<tensor<64x10xf32>>>, tensor<64x10xf32>) -> ()
                                             ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:46:59: error: Could not legalize op: tf.Placeholder
      %outputs_78, %control_79 = tf_executor.island wraps "tf.Placeholder"() {device = "", shape = #tf_type.shape<?x784>} : () -> tensor<?x784xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:46:59: error: Could not legalize op: tf_executor.yield
      %outputs_78, %control_79 = tf_executor.island wraps "tf.Placeholder"() {device = "", shape = #tf_type.shape<?x784>} : () -> tensor<?x784xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:46:59: error: Could not legalize op: tf_executor.island
      %outputs_78, %control_79 = tf_executor.island wraps "tf.Placeholder"() {device = "", shape = #tf_type.shape<?x784>} : () -> tensor<?x784xf32>
                                                          ^
/home/rashik/Documents/tensorfow_mlir_test/tf_dialect.mlir:47:59: error: Node `mhlo.dot` must have compile-time constant inputs and outputs.

XLA compilation requires that operator arguments that represent shapes or dimensions be evaluated to concrete values at compile time.  This error means that a shape or dimension argument could not be evaluated at compile time, usually because the value of the argument depends on a parameter to the computation, on a variable, or on a stateful operation such as a random number generator.
      %outputs_80, %control_81 = tf_executor.island wraps "tf.MatMul"(%outputs_78, %outputs_11) <{transpose_a = false, transpose_b = false}> {device = ""} : (tensor<?x784xf32>, tensor<784x64xf32>) -> tensor<?x64xf32>
```
