Searched refs:ExecuteOnStreamWrapper (Results 1 – 4 of 4) sorted by relevance
319 StatusOr<ScopedShapedBuffer> ExecuteOnStreamWrapper(323 StatusOr<ExecutionOutput> ExecuteOnStreamWrapper(
162 StatusOr<ScopedShapedBuffer> Executable::ExecuteOnStreamWrapper( in ExecuteOnStreamWrapper() function in xla::Executable173 StatusOr<ExecutionOutput> Executable::ExecuteOnStreamWrapper( in ExecuteOnStreamWrapper() function in xla::Executable
177 executable->ExecuteOnStreamWrapper(&service_run_options, in ExecuteWithDeviceBuffers()
506 TF_ASSIGN_OR_RETURN(auto result, executable->ExecuteOnStreamWrapper( in ExecuteAndRegisterResult()