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