tensorflow/compiler/aot/tests/test_graph_tfgather.config.pbtxt


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

# Text form of tensorflow.tf2xla.Config proto.
feed {
  id { node_name: "params" }
  shape {
    dim { size: 4 }
  }
}
feed {
  id { node_name: "indices" }
  shape {
    dim { size: 2 }
  }
}
fetch {
  id { node_name: "gather_output" }
}