| 95 | } |
| 96 | |
| 97 | std::vector<std::unique_ptr<GlobalData>> MakeFakeArgumentsOrDie( |
| 98 | const XlaComputation& computation, Client* client, |
| 99 | DebugOptions* debug_opts /*=nullptr*/) { |
| 100 | CHECK(computation.proto().has_host_program_shape()) |
| 101 | << "Computation should have program shape."; |
| 102 | auto program_shape = computation.proto().host_program_shape(); |
| 103 | |
| 104 | std::vector<std::unique_ptr<GlobalData>> results; |
| 105 | for (const ShapeProto& shape : program_shape.parameters()) { |
| 106 | results.push_back(MakeFakeDataOrDie(Shape(shape), client, debug_opts)); |
| 107 | } |
| 108 | return results; |
| 109 | } |
| 110 | |
| 111 | } // namespace xla |
no test coverage detected