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