Searched refs:XlaCompiledProgramShape (Results 1 – 1 of 1) sorted by relevance
224 xla::ProgramShape XlaCompiledProgramShape( in XlaCompiledProgramShape() function848 XlaCompiledProgramShape(xla_computation, xla::ProgramShape(*shapes)); in TEST()