std::uniform_int_distribution<int> some_operation(0, num_operations - 1);

      for (int i = 0; i < num_tensors; ++i) {
        AddTensor(SizeT{1} << some_size_log(rng));
      }
      for (int i = 0; i < num_operations; ++i) {
        AddOperation(/*is_stateful=*/false);
      }
      for (int i = 0; i < num_uses; ++i) {
        AddUse(some_operation(rng), some_tensor(rng));
      }
    }
  };
};

// Infers constraints for the values in the function body from the function
// results attributes.
//
// Example:
//   func @test(...) -> (tensor<?x?xf32> {tf.constraint = "shape"}) {
//     .....
//     %v = "some_operation"() : () -> tensor<?x?xf32>
//     return %v : tensor<?x?xf32>
//   }
LogicalResult InferFunctionBodyValuesConstraints(
    func::FuncOp func, ValuesConstraintSet& constraints);

}  // namespace TFDevice

// CHECK:   operator_codes: [ {
// CHECK:     deprecated_builtin_code: 32,
// CHECK:     custom_code: "SomeOperation",
// CHECK:     builtin_code: CUSTOM
// CHECK:   } ],
// CHECK:   subgraphs: [ {
// CHECK:     tensors: [ {
// CHECK:       shape: [  ],
// CHECK:       type: INT32,
// CHECK:       buffer: 1,
// CHECK:       name: "tf.SomeOperation",
// CHECK:       quantization: {
// CHECK-EMPTY
// CHECK:       }
// CHECK:     } ],