Home
last modified time | relevance | path

Searched refs:CompiledProgramShape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dlibtpu.h76 typedef struct CompiledProgramShape { struct
80 } CompiledProgramShape; argument
190 typedef struct CompiledProgramShape*(
195 struct CompiledProgramShape* shape);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
H A Ddirect_tpu_driver.cc157 struct CompiledProgramShape* shape = in program_shape()