Searched refs:runfn (Results 1 – 3 of 3) sorted by relevance
117 xla::StatusOr<T> Run(const std::function<xla::StatusOr<T>()>& runfn,159 const std::function<xla::StatusOr<T>()>& runfn, xla::Backend* backend, in Run() argument167 auto result_or = runfn(); in Run()
236 auto runfn = [&]() -> xla::StatusOr<xla::ExecutionOutput> { in ExecuteTPUProgram() local244 runfn, backend, device_ordinal, /*requested_free_size=*/0); in ExecuteTPUProgram()
381 auto runfn = [&]() { in ExecuteComputation() local390 runfn, device_ref->backend(), device_ref->device_ordinal(), in ExecuteComputation()