Searched refs:input_program_shape (Results 1 – 1 of 1) sorted by relevance
226 const xla::ProgramShape& input_program_shape) { in XlaCompiledProgramShape() argument232 exec_options.set_result_layout(input_program_shape.result()); in XlaCompiledProgramShape()234 for (int64 i = 0; i < input_program_shape.parameters_size(); ++i) { in XlaCompiledProgramShape()235 parameters_shapes.push_back(&input_program_shape.parameters(i)); in XlaCompiledProgramShape()