resource_ops.mlir

module  {
  flow.variable @counter mutable dense<0.000000e+00> : tensor<f32> attributes {sym_visibility = "private"} loc(#loc0)
  func @add_assign(%arg0: tensor<f32>) -> tensor<f32> attributes {iree.abi = "{\22a\22:[[\22ndarray\22,\22f32\22,0]],\22r\22:[[\22ndarray\22,\22f32\22,0]],\22v\22:1}"} {
    %0 = flow.variable.load @counter : tensor<f32> loc(#loc1)
    %1 = mhlo.add %0, %arg0 : tensor<f32> loc(#loc1)
    flow.variable.store %1, @counter : tensor<f32> loc(#loc1)
    %2 = flow.variable.load @counter : tensor<f32> loc(#loc2)
    return %2 : tensor<f32> loc(#loc0)
  } loc(#loc0)
} loc(#loc0)
#loc0 = loc(unknown)
#loc1 = loc(callsite("AssignAddVariableOp@__inference_add_assign_13"("/usr/local/google/home/laurenzo/src/iree/iree.venv/lib/python3.9/site-packages/tensorflow/python/ops/gen_resource_variable_ops.py":56:0) at callsite("/usr/local/google/home/laurenzo/src/iree/iree.venv/lib/python3.9/site-packages/tensorflow/python/ops/resource_variable_ops.py":855:0 at callsite("/usr/local/google/home/laurenzo/src/iree/integrations/tensorflow/e2e/resource_ops_test.py":29:0 at callsite("/usr/local/google/home/laurenzo/src/iree/iree.venv/lib/python3.9/site-packages/tensorflow/python/framework/func_graph.py":983:0 at callsite("/usr/local/google/home/laurenzo/src/iree/iree.venv/lib/python3.9/site-packages/tensorflow/python/eager/function.py":3983:0 at callsite("/usr/local/google/home/laurenzo/src/iree/iree.venv/lib/python3.9/site-packages/tensorflow/python/eager/def_function.py":668:0 at callsite("/usr/local/google/home/laurenzo/src/iree/iree.venv/lib/python3.9/site-packages/tensorflow/python/framework/func_graph.py":1007:0 at callsite("/usr/local/google/home/laurenzo/src/iree/iree.venv/lib/python3.9/site-packages/tensorflow/python/eager/function.py":3291:0 at callsite("/usr/local/google/home/laurenzo/src/iree/iree.venv/lib/python3.9/site-packages/tensorflow/python/eager/function.py":3456:0 at "/usr/local/google/home/laurenzo/src/iree/iree.venv/lib/python3.9/site-packages/tensorflow/python/eager/function.py":3059:0))))))))))
#loc2 = loc(callsite("ReadVariableOp@__inference_add_assign_13"("/usr/local/google/home/laurenzo/src/iree/iree.venv/lib/python3.9/site-packages/tensorflow/python/saved_model/save.py":1081:0) at callsite("/usr/local/google/home/laurenzo/src/iree/iree.venv/lib/python3.9/site-packages/tensorflow/python/saved_model/save.py":1043:0 at callsite("/usr/local/google/home/laurenzo/src/iree-build/bindings/python/iree/compiler/tf.py":212:0 at callsite("/usr/local/google/home/laurenzo/src/iree-build/bindings/python/iree/compiler/tf.py":216:0 at callsite("/usr/local/google/home/laurenzo/src/iree-build/bindings/python/iree/tf/support/module_utils.py":120:0 at callsite("/usr/local/google/home/laurenzo/src/iree-build/bindings/python/iree/tf/support/module_utils.py":369:0 at callsite("/usr/local/google/home/laurenzo/src/iree-build/bindings/python/iree/tf/support/module_utils.py":350:0 at callsite("/usr/local/google/home/laurenzo/src/iree-build/bindings/python/iree/tf/support/module_utils.py":940:0 at callsite("/usr/local/google/home/laurenzo/src/iree-build/bindings/python/iree/tf/support/tf_test_utils.py":162:0 at "/usr/local/google/home/laurenzo/src/iree-build/bindings/python/iree/tf/support/tf_test_utils.py":167:0))))))))))