Searched refs:GetCompiledProgramShape (Results 1 – 1 of 1) sorted by relevance
795 xla::Status GetCompiledProgramShape(int64_t op_id, in GetCompiledProgramShape() function in tpu_driver::__anon2989b0910111::PodTpuDriver964 return driver_->GetCompiledProgramShape(operation_id(), program_shape); in program_shape()